pytorch能自动反向传播。 前向传播时,能够自动记录每个操作的拓扑顺序,反向传播时则可以自动反向传播。
Variable
Variable是一个类,对是对tensor的一个wrapper。
有三个属性.data, .grad和.creator。(注意:新版的.creator已经改成.grad_fn了,这里不再说明。)如果是第一个运算节点,显然.creator是空,.grad只有调用.backward时候才会有。
require_grad可以指定某个variable是否计算其梯度。
import torch
from torch.autograd import Variable
x = Variable(torch.ones(2, 2), requires_grad=True)
print(x)

最低0.47元/天 解锁文章
4752

被折叠的 条评论
为什么被折叠?



