PyTorch中permute的用法詳解
更新時間：2019年12月30日 08:58:05 作者：一只想入門卻未入門的程序猿
今天小編就為大家分享一篇PyTorch中permute的用法詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
permute(dims)
將tensor的維度換位。
參數(shù)：參數(shù)是一系列的整數(shù)，代表原來張量的維度。比如三維就有0，1，2這些dimension。
例：
import torch import numpy as np a=np.array([[[1,2,3],[4,5,6]]]) unpermuted=torch.tensor(a) print(unpermuted.size()) # ——> torch.Size([1, 2, 3]) permuted=unpermuted.permute(2,0,1) print(permuted.size()) # ——> torch.Size([3, 1, 2])
再比如圖片img的size比如是（28，28，3）就可以利用img.permute(2,0,1)得到一個size為（3，28，28）的tensor。
利用這個函數(shù)permute（1，3，2）可以把Tensor([[[1,2,3],[4,5,6]]]) 轉(zhuǎn)換成
tensor([[[1., 4.], [2., 5.], [3., 6.]]])
如果使用view（1,3,2），可以得到
tensor([[[1., 2.], [3., 4.], [5., 6.]]])
以上這篇PyTorch中permute的用法詳解就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了Python with用法：自動關(guān)閉文件進程,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了python檢測遠程端口是否打開的方法,實例分析了Python基于socket檢測端口的技巧,需要的朋友可以參考下2015-03-03
在學習Python處理二維碼的過程中，我們看到的大多是“用python生成酷炫二維碼”、“用Python制作動圖二維碼”之類的文章。而關(guān)于使用Python批量識別二維碼的教程，并不多見。本文就為大家總結(jié)了Python識別二維碼的兩種方法，感興趣的可以學習一下2022-06-06
正則表達式是一種用來匹配字符串的強有力的武器,利用字符來匹配字符的思想,基于顯示規(guī)則進行模式匹配,可以高效組合成不同樣式的字符串,下面這篇文章主要給大家介紹了關(guān)于python正則表達式用法超詳細講解的相關(guān)資料,需要的朋友可以參考下2022-10-10
這篇文章主要介紹了解讀pandas.DataFrame.corrwith，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-11-11
這篇文章主要給大家介紹了關(guān)于python更換國內(nèi)鏡像源三種實用方法的相關(guān)資料,更換Python鏡像源可以幫助解決使用pip安裝包時速度過慢或無法連接的問題,需要的朋友可以參考下2023-09-09
Python如何實現(xiàn)Markdown記賬記錄轉(zhuǎn)Excel存儲
這篇文章主要為大家詳細介紹了Python如何實現(xiàn)Markdown記賬記錄轉(zhuǎn)Excel存儲功能,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下2025-03-03
本文總結(jié)了Python中from和import的使用方法,包括導入模塊和導入模塊中的對象,強調(diào)了模塊和對象的概念,并給出了具體的使用示例,最后,建議根據(jù)代碼維護性和團隊編碼風格來選擇合適的導入方式2025-03-03
這篇文章主要介紹了Python如何用pip命令升級所有可以升級的(過時的)包,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-03-03
python輾轉(zhuǎn)相除法求最大公約數(shù)和最小公倍數(shù)的實現(xiàn)
這篇文章主要介紹了python輾轉(zhuǎn)相除法求最大公約數(shù)和最小公倍數(shù)的實現(xiàn)方式，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-07-07