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