今天我们讲解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中使用Variable类封装Tensor,并展示了其基本功能如梯度跟踪、数据访问和求导方法。后续还将探讨.backward()函数的使用。理解Variable是进行梯度计算和优化的基础。
931

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



