随笔分类 -  pytorch

 
python中list array torch相互转换
摘要：一、生成array\list\tensor 1.生成array import numpy as np a1 = array([[1,2],[3,4]]) 2. 生成list a2 = list(range(16)) 3.生成tensor import torch a3 = torch.tensor( 阅读全文
posted @ 2020-12-03 16:59 dychen0408 阅读(5176) 评论(0) 推荐(0)
pytorch 中 torch.no_grad()、requires_grad、eval()
摘要：requires_grad requires_grad=True 要求计算梯度； requires_grad=False 不要求计算梯度； 在pytorch中，tensor有一个 requires_grad参数，如果设置为True，则反向传播时，该tensor就会自动求导。 tensor的requi 阅读全文
posted @ 2020-11-03 21:35 dychen0408 阅读(4278) 评论(0) 推荐(1)