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

Question

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

Answer 1

您可以使用以下命令将其强制torch.Tensor到列表中list：

>>> 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]