PyTorch-Variable变量

最新推荐文章于 2023-12-08 15:41:11 发布

lwycc233

最新推荐文章于 2023-12-08 15:41:11 发布

阅读量171

点赞数

CC 4.0 BY-SA版权

分类专栏： PyTorch

本文链接：https://blog.youkuaiyun.com/lwycc2333/article/details/88979336

PyTorch 专栏收录该内容

14 篇文章

订阅专栏

本文介绍PyTorch中Variable与Tensor的区别及转换方法，演示如何通过requires_grad参数控制反向传播，并展示如何打印Variable的梯度值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Variable和Tensor的形式转化
requires_grad参数涵义
打印Variable梯度

import torch as t
from torch.autograd import Variable

tensor = t.FloatTensor([[1,2],[3,4]])
variable = Variable(tensor,requires_grad = True) # requires_grad:是否可以反向传播
t_out = t.mean(tensor*tensor) # mean 均值
v_out = t.mean(variable*variable)
print(t_out)
print(v_out)

v_out.backward()

print(variable.grad) #打印梯度值
print(variable.data.numpy()) # variable是variable形式，variable.data是tensor形式