定义一个列表数据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)