0
点赞
收藏
分享

微信扫一扫

pytorch tensor 转 variable

未定义变量 2022-07-27 阅读 74


import torch
from torch.autograd import Variable
t = torch.zeros([2,2])
v = Variable(t)
print(t)
print("-----")
print(v)

print结果:

0  0
0 0
[torch.FloatTensor of size 2x2]

-----
Variable containing:
0 0
0 0
[torch.FloatTensor of size 2x2]


举报

相关推荐

0 条评论