pytorch實(shí)現(xiàn)Tensor變量之間的轉(zhuǎn)換
系統(tǒng)默認(rèn)是torch.FloatTensor類型
data = torch.Tensor(2,3)是一個(gè)2*3的張量，類型為FloatTensor
data.cuda()就轉(zhuǎn)換為GPU的張量類型，torch.cuda.FloatTensor類型
(1) CPU或GPU之間的張量轉(zhuǎn)換
在Tensor后加long(), int(), double(),float(),byte()等函數(shù)就能將Tensor進(jìn)行類型轉(zhuǎn)換type()函數(shù)，
data為Tensor數(shù)據(jù)類型，data.type()為給出data的類型，
如果使用data.type(torch.FloatTensor)則強(qiáng)制轉(zhuǎn)換為torch.FloatTensor類型張量
(2) CPU張量轉(zhuǎn)化成GPU張量
data.cuda()
(3) GPU張量轉(zhuǎn)化成CPU張量
data.cpu()
(4) Variable變量轉(zhuǎn)換成普通的Tensor
Variable是一個(gè)Wrapper,裝在里面的data是tensor，如果Var是Variable變量，使用Var.data獲得Tensor變量
(5) Tensor與numpy array之間的轉(zhuǎn)換
Tensor->numpy 使用data.numpy(),data為Tensor變量
Numpy->Tensor 使用torch.from_numpy(data)，data為numpy變量
(6) 分別獲取張量和數(shù)組的尺寸,注意size的使用
torch 張量
獲取張量的尺寸 a.size()
numpy 數(shù)組
獲取數(shù)組的尺寸 b.shape
獲取數(shù)組中元素的個(gè)數(shù): b.size (這里和張量中的屬性的size的含義不同)
(7) 升維和降維的問題
unsqueeze(N)升維到第N維
squeeze(N)降維第N維
需要做如下操作:
x = x.unsqueeze(0) 假如x=(3,1080,1920) 操作后 x = (1,3,1080,1920)
降維也是做同樣的操作
以上這篇pytorch實(shí)現(xiàn)Tensor變量之間的轉(zhuǎn)換就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python安裝sklearn模塊的方法詳解,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-11-11
Python中使用logging模塊代替print（logging簡(jiǎn)明指南）
這篇文章主要介紹了Python中使用logging模塊代替print的好處說明,主旨是logging模塊簡(jiǎn)明指南,logging模塊的使用方法介紹,需要的朋友可以參考下2014-07-07
前面九篇文章從基礎(chǔ)到編寫都做了詳細(xì)的介紹了，第十篇么講究個(gè)十全十美，那么我們就來詳細(xì)記錄一下一個(gè)爬蟲程序如何一步步編寫出來的，各位看官可要看仔細(xì)了2014-11-11
Python+eval函數(shù)實(shí)現(xiàn)動(dòng)態(tài)地計(jì)算數(shù)學(xué)表達(dá)式詳解
Python的 eval() 允許從基于字符串或基于編譯代碼的輸入中計(jì)算任意Python表達(dá)式。當(dāng)從字符串或編譯后的代碼對(duì)象的任何輸入中動(dòng)態(tài)計(jì)算Python表達(dá)式時(shí)，此函數(shù)非常方便。本文將利用eval實(shí)現(xiàn)動(dòng)態(tài)地計(jì)算數(shù)學(xué)表達(dá)式，需要的可以參考一下2022-09-09
基于Python實(shí)現(xiàn)打哈欠檢測(cè)詳解
這篇文章主要介紹了如何利用Python+OpenCV實(shí)現(xiàn)人打哈欠檢測(cè)，文中的示例代碼講解詳細(xì)，對(duì)我們學(xué)習(xí)Python有一定幫助，感興趣的可以了解一下2022-04-04
Python自動(dòng)掃雷實(shí)現(xiàn)方法
這篇文章主要介紹了Python自動(dòng)掃雷實(shí)現(xiàn)方法,提供了Python實(shí)現(xiàn)自動(dòng)掃雷的代碼思路并備有注釋說明,具有一定參考借鑒價(jià)值,需要的朋友可以參考下2015-07-07
首先，在認(rèn)識(shí)metaclass之前，你需要認(rèn)識(shí)下python中的class。python中class的奇怪特性借鑒了smalltalk語言。大多數(shù)語言中，classes僅僅是用于描述怎樣創(chuàng)建一個(gè)對(duì)象的代碼端。在某種程度上說，python中的class也是這樣的。2016-02-02
python 如何用 Hypothesis 來自動(dòng)化單元測(cè)試
這篇文章主要介紹了python 如何用 Hypothesis 來自動(dòng)化單元測(cè)試，幫助大家更好的理解和學(xué)習(xí)使用python，感興趣的朋友可以了解下2021-03-03
Python使用numpy模塊創(chuàng)建數(shù)組操作示例
這篇文章主要介紹了Python使用numpy模塊創(chuàng)建數(shù)組操作,結(jié)合實(shí)例形式分析了Python使用numpy模塊實(shí)現(xiàn)數(shù)組的創(chuàng)建、賦值、修改、打印等相關(guān)操作技巧與注意事項(xiàng),需要的朋友可以參考下2018-06-06
這篇文章主要介紹了python繪圖模塊matplotlib示例詳解,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-07-07