pytorch permute維度轉換方法
更新時間:2018年12月14日 15:38:42 作者:ShellCollector
今天小編就為大家分享一篇pytorch permute維度轉換方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
permute
prediction = input.view(bs, self.num_anchors, self.bbox_attrs, in_h, in_w).permute(0, 1, 3, 4, 2).contiguous()
轉置:
import torch x = torch.linspace(1, 9, steps=9).view(3, 3) b=x.permute(1,0) print(b) print(b.permute(1,0))
以上這篇pytorch permute維度轉換方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
您可能感興趣的文章:
相關文章
詳解Python3之數(shù)據(jù)指紋MD5校驗與對比
這篇文章主要介紹了Python3之數(shù)據(jù)指紋MD5校驗與對比,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2019-06-06pycharm+PyQt5+python最新開發(fā)環(huán)境配置(踩坑)
這篇文章主要介紹了pycharm+PyQt5+python最新開發(fā)環(huán)境配置(踩坑),小編覺得挺不錯的,現(xiàn)在分享給大家,也給大家做個參考。一起跟隨小編過來看看吧2019-02-02Python基礎進階之海量表情包多線程爬蟲功能的實現(xiàn)
這篇文章主要介紹了Python基礎進階之海量表情包多線程爬蟲,本文通過實例代碼給大家介紹的非常詳細,對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-12-12django實現(xiàn)模型字段動態(tài)choice的操作
這篇文章主要介紹了django實現(xiàn)模型字段動態(tài)choice的操作,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-04-04