pytorch 实现tensor与numpy数组转换

Python  /  管理员 发布于 5年前   930

看代码，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数组转换就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。