如何将装有tensor的多维list转化为torch.Tensor类型

普通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就会得到以下结果：