python - 张量列表成张量

Question

我有一个类似的清单

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

我想将其更改为张量，以便我可以将 xs 和 ys 输入 Pytorch DataLoader。我试过这个，

xs = torch.Tensor(xs)

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

标签： pythonpython-3.xlistdeep-learningpytorch

Answer 1

你可能想要torch.stack

xs = torch.stack(xs)