普通list转tensor:

a = [[j for j in range(5)] for i in range(3)]
A = torch.Tensor(a)

print('list = ', a)
print('type of a : ', type(a))
print('type of transform: ', type(A))

结果:
可以看到对于不包含tensor的普通list,直接用torch.Tensor就可以进行转换。
在这里插入图片描述

但是对于包含了tensor的list,使用torch.Tensor就会报错:

a = torch.rand((2, 3))
a_list = [a for _ in range(3)]
A = torch.Tensor(a_list)

结果:
在这里插入图片描述
可以通过torch.stack将包含tensor的多维list转换成tensor:

a = torch.rand((2, 3))
a_list = [a for _ in range(3)]
A = torch.stack(a_list)

print('list = ', a_list)
print('type of a_list : ', type(a_list))
print('size of a_list : ', [len(a_list), len(a_list[0]), len(a_list[0][0])])
print('type of transform: ', type(A))
print('shape of transform: ', A.shape)

结果:
可以看到torch.stack不仅可以把多维list转换成tensor,并且维度不变。
在这里插入图片描述
torch.cat也可以把list转换成tensor,但是它会把指定的dim直接堆砌到一起(默认dim=0),上面的stack如果换成cat就会得到以下结果:
在这里插入图片描述

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐