定义一个列表数据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)
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
