【Python秘籍】numpy到tensor的转换

来源：飒榕旅游知识分享网

在⽤pytorch训练神经⽹络时，我们常常需要在numpy的数组变量类型与pytorch中的tensor类型进⾏转换，今天给⼤家介绍⼀种它们之间互相转换的⽅法。

⼀、numpy到tensor

⾸先我们要引⼊必要的包：

import numpy as npimport torch

然后创建⼀个numpy类型的数组：

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

这⾥创建了⼀个⼀维的数组，5个都为1，我们打印⼀下这个x的类型显⽰如下：

这个就说明现在x是numpy类型的⼀个数组，接着我们⽤下⾯的代码将x转换成tensor类型：

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

这个打印的结果是：

说明我们成功的转换了！

⼆、tensor到numpy

直接上代码：

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

这⾥的x就是刚刚我们转换成tensor的那个x，打印结果如下：

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

因篇幅问题不能全部显示，请点此查看更多更全内容

查看全文