【pytorch】variable 和 tensor

最新推荐文章于 2023-01-31 10:46:22 发布

原创最新推荐文章于 2023-01-31 10:46:22 发布 · 917 阅读

0 ·

CC 4.0 BY-SA版权

pytorch 专栏收录该内容

7 篇文章

订阅专栏

本文通过PyTorch创建了一个简单的张量，并将其转化为Variable类型以进行自动梯度计算。展示了如何使用PyTorch执行基本的数学运算并进行反向传播，最终将Variable类型的数据转换为NumPy数组。

import torch
from torch.autograd import Variable

tensor = torch.FloatTensor([[1,2],[3,4]])
variable = Variable(tensor,requires_grad=True)

print(tensor)
print(variable)

t_out = torch.mean(tensor * tensor)
v_out = torch.mean(variable * variable)

v_out.backward()
print('反向传播之后',variable) #反向传播的时候variable 是受影响的，因为反向传播了

print('变量的data是tensor类型--',variable.data)
print('variable转为numpy--',variable.data.numpy())# 必须借助于tensor转换为numpy