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设计】。

返回列表

基础版(list方法) # 比较占内存 w = int(input("输入一个数字还你一个斐波那契数列：")) list_res = [] def list_n(n): if...

按照Python官网上的计划，Python3.6正式版期望在2016-12-16号发布，也就是这周五。从去年的5月份开始，Python3.6版本就已经动手开发了，期间也断断续续的发布了4...

本文实例讲述了Python实现图像几何变换的方法。分享给大家供大家参考。具体实现方法如下： import Image try: im=Image.open('test.jpg')...

本文实例讲述了Python可变参数函数用法。分享给大家供大家参考。具体如下： #!/usr/bin/python def f1(a,b): print a,b def f2(a,*b...

本例设置为垂直左侧scroll 主要思想是利用一个长度为0的mid_frame，高度为待设置qwidget的高度，用mid_frame的moveEvent事件驱动qwidget的move...

宜配屋