list，ndarray，tensor之间转换

zihan_2024

已于 2024-10-29 23:43:58 修改

阅读量663

点赞数 1

文章标签： python list

于 2021-08-14 12:18:03 首次发布

本文链接：https://blog.youkuaiyun.com/qq_40639828/article/details/119699156

版权

本文介绍了如何在Python环境中使用numpy和torch进行不同类型数据（list、ndarray和tensor）之间的转换，并展示了转换过程中的实例。重点在于理解list到ndarray和tensor，以及ndarray和tensor之间的相互转换操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

环境：python3.7.10、numpy1.21.0、torch1.5.0+cu101

import numpy as np
import torch

# 定义三种类型数据
list = [[1,2,3],[2,3,4]]
ndarray = np.array([[1,2,3],[2,3,4]])
tensor = torch.tensor([[1,2,3],[2,3,4]])

print(list, type(list))
"""[[1, 2, 3], [2, 3, 4]] <class 'list'>"""

print(ndarray, type(ndarray))
"""[[1 2 3]
    [2 3 4]] <class 'numpy.ndarray'>"""

print(tensor, type(tensor))
"""tensor([[1, 2, 3],
           [2, 3, 4]]) <class 'torch.Tensor'>"""

# list to ndarray and Tensor
ndarray = np.array(list)
tensor = torch.tensor(list)
print(ndarray, "\n", tensor)
"""
[[1 2 3]
 [2 3 4]] 
 tensor([[1, 2, 3],
        [2, 3, 4]])
"""

# ndarray to list and Tensor
list = ndarray.tolist()
tensor = torch.from_numpy(ndarray)
print(list, "\n", tensor)
"""
[[1, 2, 3], [2, 3, 4]] 
 tensor([[1, 2, 3],
        [2, 3, 4]], dtype=torch.int32)
"""

# Tensor to ndarray and list
list = tensor.numpy().tolist()
ndarray = tensor.numpy()
print(list, "\n", ndarray)

"""
[[1, 2, 3], [2, 3, 4]] 
 [[1 2 3]
 [2 3 4]]
"""