如何将张量转换为张量列表

Question

我该如何将一个张量转换为张量列表？例如：P1 是一个包含 60 个值的 torch.Tensor，我想要一个包含 60 个张量的张量列表。

- Saran Zeb

---

你是否在寻找一种更优雅的方法，而不需要使用循环？因为你可以只需循环遍历并将它们附加到一个空列表中。 - Natthaphon Hongcharoen

Answer 1

你可以使用list将torch.Tensor强制转换为列表:

>>> P1 = torch.rand(60)
>>> list(P1)
[tensor(0.5987),
 tensor(0.5321),
 tensor(0.6590),
 ...
 tensor(0.1381)]

这也适用于多维张量：

>>> P1 = torch.rand(60, 2)
>>> list(P1)
[tensor([0.4675, 0.0430]),
 tensor([0.2175, 0.6271]),
 tensor([0.3378, 0.8516]),
 ...,
 tensor([0.5099, 0.3411]

- Ivan