Pytorch中的Variable

Pytorch中的Variable,第1张

Pytorch中的Variable

pytorch两个基本对象:Tensor(张量)和Variable(变量)
其中,tensor不能反向传播,variable可以反向传播。



Varibale包含三个属性:
●data: 存储了Tensor, 是本体的数据
●grad: 保存了data的梯度, 本事是个Variable而非Tensor, 与data形状一致
●grad_ fn:指向Function对象, 用于反向传播的梯度计算之用,表示是通过什么 *** 作得到这个变量的例如( 加减乘除、卷积、反置卷积)
用法:

  1. import torch
  2. from torch.autograd import Variable
  3. x = Variable(torch.one(2,2), requires_ grad= True)
  4. print(x)#其实查询的是x.data,是个tensor

欢迎分享,转载请注明来源:内存溢出

原文地址:https://www.54852.com/langs/579024.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-04-11
下一篇2022-04-11

发表评论

登录后才能评论

评论列表(0条)

    保存