pytorch 实现tensor与numpy数组转换
作者:头发光了你就强了
今天小编就为大家分享一篇使用pytorch 实现tensor与numpy数组转换,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
看代码,tensor转numpy:
a = torch.ones(2,2) b = a.numpy() c=np.array(a) #也可以转numpy数组 print(type(a)) print(type(b)) print(a) print(b)
输出为:
<class ‘torch.Tensor'> <class ‘numpy.ndarray'> tensor([[1., 1.], [1., 1.]]) [[1. 1.] [1. 1.]]
numpy转tensor:
import torch import numpy as np a = np.ones(5) b = torch.from_numpy(a) c=torch.Tensor(a) #也可以转pytorch Tensor print(type(a)) print(type(b)) print(a) print(b)
输出为:
<class ‘numpy.ndarray'> <class ‘torch.Tensor'> [1. 1. 1. 1. 1.] tensor([1., 1., 1., 1., 1.], dtype=torch.float64)
可见pytorch的tensor对象与numpy数组是可以相互转换的,且numpy数组的默认类型是double
以上这篇pytorch 实现tensor与numpy数组转换就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。