將pytorch轉(zhuǎn)成longtensor的簡單方法
更新時間：2020年02月18日 09:00:31 作者：昕晴
今天小編就為大家分享一篇將pytorch轉(zhuǎn)成longtensor的簡單方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
我就廢話不多說了，直接上代碼了。非常簡單哦！
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個視頻，16個關(guān)鍵幀，3通道，長為2寬為2
# 即一個視頻，有16張 3通道，長為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的簡單方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python 使用pygame工具包實(shí)現(xiàn)貪吃蛇游戲(多彩版)
這篇文章主要介紹了python 使用pygame工具包實(shí)現(xiàn)貪吃蛇游戲,本篇給大家分享的是一個多彩版，通過實(shí)例代碼給大家講解，非常詳細(xì)，具有一定的參考借鑒價值,需要的朋友可以參考下2019-10-10
今天小編就為大家分享一篇python判斷完全平方數(shù)的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11
這篇文章詳細(xì)介紹了Python sys.path，有需要的朋友可以參考一下2013-10-10
Python實(shí)現(xiàn)mysql數(shù)據(jù)庫中的SQL文件生成和導(dǎo)入
這篇文章主要介紹了Python實(shí)現(xiàn)mysql數(shù)據(jù)庫中的SQL文件生成和導(dǎo)入，首先通過將mysql數(shù)據(jù)導(dǎo)出到SQL文件中展開詳細(xì)內(nèi)容需要的小伙伴可以參考一下2022-06-06
Python requests.post方法中data與json參數(shù)區(qū)別詳解
這篇文章主要介紹了Python requests.post方法中data與json參數(shù)區(qū)別詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-04-04
這篇文章主要介紹了python statsmodel使用的相關(guān)資料，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2020-12-12
pyautogui一個神奇的圖像自動化庫,學(xué)會之后無所不能,下面這篇文章主要給大家介紹了關(guān)于python編寫腳本之pyautogui的安裝和使用的相關(guān)資料,需要的朋友可以參考下2021-12-12
這篇文章主要介紹了在python中的socket模塊使用代理實(shí)例,調(diào)用socks.setdefaultproxy即可實(shí)現(xiàn),需要的朋友可以參考下2014-05-05
在本篇文章里小編給各位分享了關(guān)于python怎么安裝scipy的具體方法和實(shí)例代碼，需要的朋友們學(xué)習(xí)下。2019-06-06
Python這個編程語言雖然沒有指針類型，但是Python中的可變參量也可以像指針一樣，改變一個數(shù)值之后，所有指向該數(shù)值的可變參量都會隨之而改變，這篇文章主要介紹了Python3中的“指針”,需要的朋友可以參考下2024-02-02