pytorch numpy list類型之間的相互轉(zhuǎn)換實例
更新時間:2019年08月18日 09:28:04 作者:WYXHAHAHA123
今天小編就為大家分享一篇pytorch numpy list類型之間的相互轉(zhuǎn)換實例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
如下所示:
import torch from torch.autograd import Variable import numpy as np ''' pytorch中Variable與torch.Tensor類型的相互轉(zhuǎn)換 ''' # 1.torch.Tensor轉(zhuǎn)換成Variablea=torch.randn((5,3)) b=Variable(a) print('a',a.type(),a.shape) print('b',type(b),b.shape) # 2.Variable轉(zhuǎn)換成torch.Tensor c=b.data#通過 Variable.data 方法相當于將Variable中的torch.tensor 取出來 print('c',c.type(),c.shape) ''' torch.tensor與numpy之間的相互轉(zhuǎn)換 ''' # 3.torch.tensor轉(zhuǎn)換成numpy d=c.numpy() # 4.numpy轉(zhuǎn)換成torch.tensor e=torch.from_numpy(d) print('d',type(d)) print('e',type(e)) ''' numpy和list之間的相互轉(zhuǎn)換 注意這種轉(zhuǎn)換只支持one-dimension array ''' # 5.numpy轉(zhuǎn)換成list f1=d.tolist() f2=list(d) # 6.list轉(zhuǎn)換成numpy g=np.asarray(f2) print('f1',type(f1)) print('f2',type(f2)) print('g',type(g)) ''' a torch.FloatTensor torch.Size([5, 3]) b <class 'torch.Tensor'> torch.Size([5, 3]) c torch.FloatTensor torch.Size([5, 3]) d <class 'numpy.ndarray'> e <class 'torch.Tensor'> f1 <class 'list'> f2 <class 'list'> g <class 'numpy.ndarray'> '''
以上這篇pytorch numpy list類型之間的相互轉(zhuǎn)換實例就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
python中for語句簡單遍歷數(shù)據(jù)的方法
這篇文章主要介紹了python中for語句簡單遍歷數(shù)據(jù)的方法,以一個簡單實例形式分析了Python中for語句遍歷數(shù)據(jù)的技巧,具有一定參考借鑒價值,需要的朋友可以參考下2015-05-05Python使用try except處理程序異常的三種常用方法分析
這篇文章主要介紹了Python使用try except處理程序異常的三種常用方法,結(jié)合實例形式分析了Python基于try except語句針對異常的捕獲、查看、回溯等相關(guān)操作技巧,需要的朋友可以參考下2018-09-09python機器學習算法與數(shù)據(jù)降維分析詳解
這篇文章主要為大家介紹了python機器學習算法與數(shù)據(jù)降維的分析詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步2021-11-11Python中利用Scipy包的SIFT方法進行圖片識別的實例教程
SIFT算法可以檢測圖片中的局部特征,算法原理相當復(fù)雜...但是!Python強大的第三方包Scipy中帶有實現(xiàn)SIFT算法的SIFT方法,我們只要拿來用就可以了,下面就為大家?guī)鞵ython中利用Scipy包的SIFT方法進行圖片識別的實例教程.2016-06-06