今天小编就为大家分享一篇使用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) 输出为： tensor([[1., 1.], [1., 1.]]) [[1. 1.] [1. 1.]] numpy转tensor： import torch import numpy as np a = np.ones(5) b = t