pytorch permute维度转换方法

yipeiwu_com7年前 (2020-03-06)Python基础

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维度转换方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。

返回列表

如下所示： a = [1, 2, 3, 8, 9] def printlist(l, index): if index == len(l): return else:...

1. 查找图像中出现的人脸代码示例： #导入face_recognition模块 import face_recognition #将jpg文件加载到numpy数组中...

最近正好在寻求一种Python的数据库ORM (Object Relational Mapper)，SQLAlchemy (项目主页)这个开源项目进入了我的视线，本来想尝试着使用Djan...

python ### 刚学了python不久，发现了一个好玩的库pygame 使用pygame模块利用面向对象的思想编写贪吃蛇，主要用到pygame.sprite：游戏主类 im...

以前游戏2048火的时候，正好用其他的语言编写了一个，现在学习python，正好想起来，便决定用python写一个2048，由于没学过python里面的界面编程，所以写了一个极其简单的无...

宜配屋