將pytorch轉(zhuǎn)成longtensor的簡(jiǎn)單方法
更新時(shí)間：2020年02月18日 09:00:31 作者：昕晴
今天小編就為大家分享一篇將pytorch轉(zhuǎn)成longtensor的簡(jiǎn)單方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
我就廢話不多說(shuō)了，直接上代碼了。非常簡(jiǎn)單哦！
pytorch轉(zhuǎn)成longtensor
b = torch.rand(3,3)#得到的是floattensor值， b = b.long()#得到的是longtensor值
pytorch tensor轉(zhuǎn)換實(shí)例
import torch
import numpy as np
#(1,3,16,2,2)
# 1個(gè)視頻，16個(gè)關(guān)鍵幀，3通道，長(zhǎng)為2寬為2
# 即一個(gè)視頻，有16張 3通道，長(zhǎng)為2寬為2 的圖像
data = np.array([[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3]
,[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3]
,[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3]
,[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3],[1,1,1,1,2,2,2,2,3,3,3,3]])
print data
torch_data = torch.FloatTensor(data) # 轉(zhuǎn)tensor
print torch_data
b = torch_data.view(1,3,16,2,2)
print b
以上這篇將pytorch轉(zhuǎn)成longtensor的簡(jiǎn)單方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
使用Python的PIL庫(kù)給圖像進(jìn)行過(guò)濾
PIL是一個(gè)用于圖像處理的Python庫(kù),它提供了各種功能,包括加載、保存、編輯和處理圖像,你可以使用PIL庫(kù)進(jìn)行圖像縮放、裁剪、旋轉(zhuǎn)、濾鏡應(yīng)用等操作,本文將介紹如何使用Python的PIL庫(kù)給圖像進(jìn)行過(guò)濾,需要的朋友可以參考下2023-08-08
Jupyter Notebook內(nèi)使用argparse報(bào)錯(cuò)的解決方案
這篇文章主要介紹了在Jupyter Notebook內(nèi)使用argparse報(bào)錯(cuò)的解決方案，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2021-06-06
Python?中用多種方式實(shí)現(xiàn)單例模式
單例模式是一種常用的軟件設(shè)計(jì)模式，該模式的主要目的是確保某一個(gè)類只有一個(gè)實(shí)例存在，本文給大家分享Python?實(shí)現(xiàn)單例模式的五種寫法，感興趣的朋友一起看看吧2022-11-11
這篇文章主要為大家詳細(xì)介紹了Python+OpenCV采集本地?cái)z像頭的視頻，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-04-04
python實(shí)現(xiàn)簡(jiǎn)易聊天室(Linux終端)
這篇文章主要為大家詳細(xì)介紹了Linux終端上python實(shí)現(xiàn)簡(jiǎn)易聊天室功能，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-07-07
這篇文章主要介紹了Keras 使用 Lambda層詳解，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-06-06
不同對(duì)象比較是按照 type names 比較，當(dāng)相同類型對(duì)象不支持適當(dāng)比較的時(shí)候，采用 address 比較,需要的朋友可以參考下2014-08-08
使用Python構(gòu)建一個(gè)簡(jiǎn)單的任務(wù)管理器
在現(xiàn)代軟件開(kāi)發(fā)中,任務(wù)管理是一個(gè)至關(guān)重要的功能,這篇文章主要為大家介紹了如何使用Python構(gòu)建一個(gè)簡(jiǎn)單的任務(wù)管理器,有需要的可以參考下2024-11-11
這篇文章主要介紹了Pytest之測(cè)試命名規(guī)則的使用，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-04-04
對(duì)Python中9種生成新對(duì)象的方法總結(jié)
今天小編就為大家分享一篇對(duì)Python中9種生成新對(duì)象的方法總結(jié)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-05-05