將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的簡單方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
PIL是一個用于圖像處理的Python庫,它提供了各種功能,包括加載、保存、編輯和處理圖像,你可以使用PIL庫進行圖像縮放、裁剪、旋轉、濾鏡應用等操作,本文將介紹如何使用Python的PIL庫給圖像進行過濾,需要的朋友可以參考下2023-08-08
Jupyter Notebook內使用argparse報錯的解決方案
這篇文章主要介紹了在Jupyter Notebook內使用argparse報錯的解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-06-06
單例模式是一種常用的軟件設計模式，該模式的主要目的是確保某一個類只有一個實例存在，本文給大家分享Python?實現(xiàn)單例模式的五種寫法，感興趣的朋友一起看看吧2022-11-11
這篇文章主要為大家詳細介紹了Python+OpenCV采集本地攝像頭的視頻，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-04-04
這篇文章主要為大家詳細介紹了Linux終端上python實現(xiàn)簡易聊天室功能，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-07-07
這篇文章主要介紹了Keras 使用 Lambda層詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06
不同對象比較是按照 type names 比較，當相同類型對象不支持適當比較的時候，采用 address 比較,需要的朋友可以參考下2014-08-08
在現(xiàn)代軟件開發(fā)中,任務管理是一個至關重要的功能,這篇文章主要為大家介紹了如何使用Python構建一個簡單的任務管理器,有需要的可以參考下2024-11-11
這篇文章主要介紹了Pytest之測試命名規(guī)則的使用，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-04-04
今天小編就為大家分享一篇對Python中9種生成新對象的方法總結，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05