pytorch tensor 转 variable

阅读 80

2022-07-27


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)

0 0 举报