看代码，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 torchimport numpy as npa = np.ones(5)b = torch.from_numpy(a)c=torch.Tensor(a) #也可以转pytorch Tensorprint(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数组转换就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。
最新文章
热门文章
猜你喜欢
