【Python秘籍】numpy到tensor的轉換

在用pytorch訓練神經網絡時，我們常常需要在numpy的數組變量類型與pytorch中的tensor類型進行轉換，今天給大家介紹一種它們之間互相轉換的方法。

一、numpy到tensor

首先我們要引入必要的包：

import numpy as np
import torch

然后創建一個numpy類型的數組：

x = np.ones(5)
print(type(x))

 

這里創建了一個一維的數組，5個都為1，我們打印一下這個x的類型顯示如下：

<class 'numpy.ndarray'>

這個就說明現在x是numpy類型的一個數組，接着我們用下面的代碼將x轉換成tensor類型：

x = torch.tensor(x)
print(type(x))

這個打印的結果是：

<class 'torch.Tensor'>

說明我們成功的轉換了！

二、tensor到numpy

直接上代碼：

x = x.detach().numpy()
print(type(x))

這里的x就是剛剛我們轉換成tensor的那個x，打印結果如下：

<class 'numpy.ndarray'>

我們就這樣成功地將他轉換回來了~