将张量列表转换为张量pytorch的张量

Question

我有此代码：

import torch

list_of_tensors = [ torch.randn(3), torch.randn(3), torch.randn(3)]
tensor_of_tensors = torch.tensor(list_of_tensors)

我遇到了错误：

ValueError：只能将一个元素张量转换为Python标量

如何在pytorch中将张量列表转换为张量张量？

Answer 1

这是一个解决方案：

tensor_of_tensors = torch.stack((list_of_tensors))
print(tensor_of_tensors) #shape (3,3)