每天讲解一点PyTorch 16Variable 三个属性 .grad .data .grad_fn
Posted knowform
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了每天讲解一点PyTorch 16Variable 三个属性 .grad .data .grad_fn相关的知识,希望对你有一定的参考价值。
今天我们讲解Variable,Variable是对Tensor的封装
from torch.autograd import Variable
x = torch.from_numpy(np.ones([1, 1, 36], dtype=np.bool)).cuda()
y =Variable(x,requires_grad=True)
#然后支持以下函数功能
y.grad
y.data
y.grad_fn #求梯度方法
后面我们计划讲解.backward(retain_graph=True)
以上是关于每天讲解一点PyTorch 16Variable 三个属性 .grad .data .grad_fn的主要内容,如果未能解决你的问题,请参考以下文章