Pytorch 如何所以Variable,Pytorch调用torch.autograd工具包

pytorch的自动求导工具包在torch.autograd中
from torch.autograd import Variable
x = torch.rand(5)
x = Variable(x,requires_grad = True)
y = x * 2
grads = torch.FloatTensor([1,2,3,4,5])
y.backward(grads)#如果y是scalar的话,那么直接y.backward(),然后通过x.grad方式,就可以得到var的梯度
x.grad #如果y不是scalar,那么只能通过传参的方式给x指定梯度
输出
Variable containing:
2
4
6
8
10
[torch.FloatTensor of size 5]

0 个评论

要回复文章请先登录注册