pytorch教程
1.梯度计算
官方:如果需要计算导数,可以在Tensor上调用.backward()。
1. 如果Tensor是一个标量(即它包含一个元素的数据),则不需要为backward()指定任何参数
2. 但是如果它有更多的元素,则需要指定一个gradient参数,它是形状匹配的张量。
import torch
from torch.autograd import Variable
x = Variable(torch.ones(2,2),requires_grad = True)
y =...
原创
2020-06-05 11:33:49 ·
237 阅读 ·
0 评论