python中的列表、数组以及张量数据类型之间的相互转换

定义一个列表数据L，可以理解为其中包含两个元素，每个元素是一个5维的向量

import torch
import numpy

L=[[1,2,3,4,5],[2,3,4,5,6]]
a=torch.Tensor(L) #列表转换为张量
b=numpy.array(L)  #列表转换为数组
b1=b.tolist()     #数组转换为列表
c=torch.from_numpy(b) #数组转换为张量
c1=c.numpy()          #张量转换为数组

print(a.size()) #张量维度
print(b.shape)  #数组大小
print(len(b1))  #列表大小
print(c.size()) #张量维度
print(c1.shape) #数组大小

为验证是否成功转换，对应输出如下： 

torch.Size([2, 5])
(2, 5)
2
torch.Size([2, 5])
(2, 5)