python - 将 PyTorch 张量转换为 python 列表

Question

如何将 PyTorch Tensor 转换为 python list？

我想将大小为 [1, 2048, 1, 1] 的张量转换为包含 2048 个元素的列表。我的张量有浮点值。是否有一种解决方案也适用于其他数据类型，例如 int？

Answer 1

使用 Tensor.tolist()例如:

>>> import torch
>>> a = torch.randn(2, 2)
>>> a.tolist()
[[0.012766935862600803, 0.5415473580360413],
 [-0.08909505605697632, 0.7729271650314331]]
>>> a[0,0].tolist()
0.012766935862600803

要删除大小为 1 的所有维度，请使用 a.squeeze().tolist()。

或者，如果除了一个维度之外的所有维度的大小都是 1(或者您希望获得张量的每个元素的列表)，您可以使用 a.flatten().tolist() .