美文网首页
【函数学习】torch.Variable()

【函数学习】torch.Variable()

作者: 风萧萧兮水易寒 | 来源:发表于2020-02-14 16:47 被阅读0次

    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

    感谢!

    相关文章

      网友评论

          本文标题:【函数学习】torch.Variable()

          本文链接:https://www.haomeiwen.com/subject/uucjfhtx.html