快捷導(dǎo)航

將PyTorch?Tensor轉(zhuǎn)換為Python列表的三種方法

更新時間：2026年04月17日 08:26:02 作者：detayun

本文介紹了三種將PyTorch Tensor轉(zhuǎn)換為Python列表的方法,推薦使用.tolist()方法,適用于任何維度的Tensor,自動處理CPU/GPU轉(zhuǎn)換且性能更優(yōu),同時提到了注意事項,.list()僅適用于一維Tensor,轉(zhuǎn)換后的列表元素保持原有的數(shù)據(jù)類型

方法一：使用.tolist()方法（推薦）

import torch

# 創(chuàng)建一個 tensor
tensor = torch.tensor([9.6919, -0.6950, 11.8760, 1.6362, 8.3674, 9.2179])

# 轉(zhuǎn)換為列表
list_result = tensor.tolist()

print(list_result)
print(type(list_result))
# 輸出: [9.6919, -0.695, 11.876, 1.6362, 8.3674, 9.2179]
# 輸出: <class 'list'>

方法二：使用list()函數(shù)

import torch

tensor = torch.tensor([9.6919, -0.6950, 11.8760, 1.6362, 8.3674, 9.2179])

# 轉(zhuǎn)換為列表
list_result = list(tensor)

print(list_result)
print(type(list_result))
# 輸出: [9.6919, -0.695, 11.876, 1.6362, 8.3674, 9.2179]
# 輸出: <class 'list'>

方法三：使用.numpy()方法（適用于 CPU tensor）

import torch

tensor = torch.tensor([9.6919, -0.6950, 11.8760, 1.6362, 8.3674, 9.2179])

# 先轉(zhuǎn)換為 numpy 數(shù)組，再轉(zhuǎn)換為列表
list_result = tensor.numpy().tolist()

print(list_result)
print(type(list_result))

處理 GPU Tensor

如果 tensor 在 GPU 上，需要先移動到 CPU：

import torch

# 假設(shè) tensor 在 GPU 上
tensor_gpu = torch.tensor([9.6919, -0.6950, 11.8760, 1.6362, 8.3674, 9.2179]).cuda()

# 先移動到 CPU，再轉(zhuǎn)換為列表
list_result = tensor_gpu.cpu().tolist()

print(list_result)

處理多維 Tensor

import torch

# 二維 tensor
tensor_2d = torch.tensor([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]])

# 轉(zhuǎn)換為嵌套列表
list_2d = tensor_2d.tolist()
print(list_2d)
# 輸出: [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]

# 三維 tensor
tensor_3d = torch.tensor([[[1.0, 2.0], [3.0, 4.0]], [[5.0, 6.0], [7.0, 8.0]]])
list_3d = tensor_3d.tolist()
print(list_3d)
# 輸出: [[[1.0, 2.0], [3.0, 4.0]], [[5.0, 6.0], [7.0, 8.0]]]

完整示例

import torch

def tensor_to_list(tensor):
    """
    將 PyTorch tensor 轉(zhuǎn)換為 Python 列表
    """
    if tensor.is_cuda:
        # 如果在 GPU 上，先移動到 CPU
        tensor = tensor.cpu()
    
    return tensor.tolist()

# 測試
tensor = torch.tensor([9.6919, -0.6950, 11.8760, 1.6362, 8.3674, 9.2179])
result = tensor_to_list(tensor)

print("原始 tensor:", tensor)
print("轉(zhuǎn)換后的列表:", result)
print("列表類型:", type(result))
print("列表元素類型:", type(result[0]))