张量到张量的列表

Question

我有一个类似

的列表

xs = [tensor1, tensor2, tensor3....]

我想将其更改为张量，以便可以将xs和ys馈入Pytorch DataLoader。 我尝试过了

xs = torch.Tensor(xs)

但是它不起作用，因为单个元素不是项目而是张量。

Answer 1

您可能需要torch.stack

xs = torch.stack(xs)