浅谈pytorch和Numpy的区别以及相互转换方法
如下所示:
#-*-coding:utf-8-*- #@Time:2018/1/1716:37 #@Author:ZhiweiZhong #@Site: #@File:Numpy_Pytorch.py #@Software:PyCharm importtorch importnumpyasnp np_data=np.arange(6).reshape((2,3)) #numpy转为pytorch格式 torch_data=torch.from_numpy(np_data) print( '\nnumpy',np_data, '\ntorch',torch_data, ) ''' numpy[[012] [345]] torch 012 345 [torch.LongTensorofsize2x3] ''' #torch转为numpy tensor2array=torch_data.numpy() print(tensor2array) """ [[012] [345]] """ #运算符 #abs、add、和numpy类似 data=[[1,2],[3,4]] tensor=torch.FloatTensor(data)#转为32位浮点数,torch接受的都是Tensor的形式,所以运算前先转化为Tensor print( '\nnumpy',np.matmul(data,data), '\ntorch',torch.mm(tensor,tensor)#torch.dot()是点乘 ) ''' numpy[[710] [1522]] torch 710 1522 [torch.FloatTensorofsize2x2] '''
以上这篇浅谈pytorch和Numpy的区别以及相互转换方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。