【Python】torch.Tensor、numpy.ndarray、list三者之间的相互转换

torch.Tensor➡numpy.ndarray

ndarray = tensor.numpy()

>>> import torch
>>> a = torch.Tensor([1])
>>> a
tensor([1.])
>>> type(a)

>>> b = a.numpy()
>>> b
array([1.], dtype=float32)
>>> type(b)

torch.Tensor➡list

list = tensor.tolist()

>>> import torch
>>> a = torch.Tensor([1])
>>> a
tensor([1.])
>>> type(a)

>>> b = a.tolist()
>>> b
[1.0]
>>> type(b)

numpy.ndarray➡torch.Tensor

tensor = torch.from_numpy(ndarray)

>>> import numpy as np
>>> import torch
>>> a = np.array([1])
>>> a
array([1])
>>> type(a)

>>> b = torch.from_numpy(a)
>>> b
tensor([1], dtype=torch.int32)
>>> type(b)

numpy.ndarray➡list

list = ndarray.tolist()

>>> import numpy as np
>>> a = np.array([1])
>>> a
array([1])
>>> type(a)

>>> b = a.tolist()
>>> b
[1]
>>> type(b)

list➡torch.Tensor

tensor = torch.Tensor(list)

>>> import torch
>>> a = [1]
>>> a
[1]
>>> type(a)

>>> b = torch.Tensor(a)
>>> b
tensor([1.])
>>> type(b)

list➡numpy.ndarray

ndarray = np.array(list)

>>> import numpy as np
>>> a = [1]
>>> a
[1]
>>> type(a)

>>> b = np.array(a)
>>> b
array([1])
>>> type(b)