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實現(xiàn)遺傳算法(二進制編碼)求函數(shù)最優(yōu)值方式
今天小編就為大家分享一篇Python實現(xiàn)遺傳算法(二進制編碼)求函數(shù)最優(yōu)值方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
python自帶了日志模塊logging，可以用來記錄程序運行過程中的日志信息。這篇文章主要介紹了python logbook使用方法,需要的朋友可以參考下2019-09-09
本文主要介紹了python實現(xiàn)矩陣的示例代碼，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-07-07
這篇文章主要介紹了Apache部署Django項目圖文詳解,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2019-07-07
這篇文章主要介紹了Python通過90行代碼搭建一個音樂搜索工具,需要的朋友可以參考下2015-07-07
今天小編就為大家分享一篇對Python中9種生成新對象的方法總結(jié)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
關(guān)于tf.reverse_sequence()簡述
今天小編就為大家分享一篇關(guān)于tf.reverse_sequence()簡述，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01
這篇文章主要介紹了python關(guān)閉windows進程的方法,涉及Python調(diào)用系統(tǒng)命令操作windows進程的技巧,需要的朋友可以參考下2015-04-04
這篇文章主要介紹了python常用命令行選項與環(huán)境變量，本文給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2020-02-02
在matplotlib中,記號是圖形兩個軸上的小標記,到目前為止,我們讓matplotlib處理軸圖例上記號的位置,下面這篇文章主要給大家介紹了關(guān)于Matplotlib控制坐標軸刻度間距與標簽的相關(guān)資料,需要的朋友可以參考下2021-10-10