torch.autograd.Variable [source] :自动微分变量,用于构建计算图
是Autograd的核心类,浅封装(thin wrapper)了Tensor,用于整合实现反向传播。torch0.4后张量与自动微分变量整合,tensor直接当作自动微分变量使用,旦Variable仍可使用。
用法:
from torch.autograd import Variable
x = Variable(tensor, requires_grad = True)
参数:
tensor:要封装的张量
requires_grad:bool,是否要求计算导数
属性:
data:tensor张量,数据本体
grad:存储的导数值,保存了data的梯度,与data形状一致
grad_fn:计算图箭头,指向Function对象,用于反向传播的梯度计算之用。旧版本中为creator
参考与详细:
官方文档 https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch-autograd/
博 客 https://blog.csdn.net/u012370185/article/details/94391428
感谢!
网友评论