list，ndarray，tensor之间转换

环境：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]]
"""