张量转换成列表 pytorch

在 PyTorch 中，可以使用 tensor.tolist() 方法将张量转换为列表。例如：

import torch

# 创建一个张量
tensor = torch.tensor([[1, 2], [3, 4]])

# 将张量转换为列表
lst = tensor.tolist()

print(lst)  # 输出：[[1, 2], [3, 4]]

注意，如果张量中包含浮点数或其他数据类型，转换后的列表中仍然是相应的数据类型。例如，如果张量中包含浮点数，则转换后的列表中仍然是浮点数。