将PyTorch张量转换为python列表

如何将PyTorch

Tensor

转换为python列表

我当前的用例是将大小为

[12048,1,1]

的张量转换为2048个元素的列表

我的张量有浮点值。是否有一种解决方案同时考虑int和其他可能的数据类型？

使用例如：

要删除尺寸

1

的所有尺寸，请使用

a.squeeze（）.tolist（）

或者，如果除一个维度外，所有维度的大小均为

1

（或者您希望获得张量的每个元素的列表），则可以使用。

要列出的张量：

a_list  = embeddings.tolist()

列表到张量：

a_tensor = torch.Tensor(a_list).cuda()

a_tensor = torch.Tensor(a_list).cuda()