目錄

• python numpy.arry, pytorch.Tensor及原生list相互轉換
  • 1 原生list轉numpy list
  • 2 numpy.array 轉原生list
  • 3 numpy.array轉torch.Tensor
  • 4 torch.Tensor轉numpy.array
  • 5 原生list轉torch.Tensor
  • 6 torch.Tensor轉原生list

python numpy.arry, pytorch.Tensor及原生list相互轉換

1 原生list轉numpy list

my_list = np.ndarray(my_list)

2 numpy.array 轉原生list

my_list = my_list.tolist()

3 numpy.array轉torch.Tensor

my_list = torch.from_numpy(my_list)

4 torch.Tensor轉numpy.array

my_list = my_list.numpy() # cpu
my_list = my_list.cpu().numpy() # gpu

5 原生list轉torch.Tensor

my_list = torch.Tensor(my_list)

6 torch.Tensor轉原生list

my_list = my_list.numpy().tolist()