【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'>

我们就这样成功地将他转换回来了~