PyTorch中Variable的用法和注意事项

用法注意事项

PyTorch中的Variable是一种封装Tensor的容器，可以跟踪其上的操作，从而实现自动求导。它的使用方法如下：

# 将Tensor封装成Variable
x = Variable(torch.Tensor([1,2,3]))

# 从已有的Variable创建
x = Variable(x.data, requires_grad=True)

# 运算
y = x + 2
z = y * y * 3
out = z.mean()

# 求导
out.backward()

# 查看梯度
x.grad

展开阅读全文

上一篇：Linux使用zip命令打包文件的简单操作步骤下一篇：什么是程序计数器？简单易懂的解释