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

返回列表

具体代码如下所示： from operator import itemgetter #itemgetter用来去dict中的key，省去了使用lambda函数 from iterto...

如何在PYTHON里运用私有属性和方法 class File: def __init__(self, name): self.name = name sel...

题目：汉诺塔给出最优解，如果对汉诺塔的定义有不了解，请翻看数据结构教材。除了最基本的之外，还有一题，给定一个数组，arr=[2,3,1,2,3]，其含义是这是一个有5个圆盘的汉诺塔...

整理总结一下python中最常用的一些时间戳和时间格式的转换第一部分：获取当前时间和10位13位时间戳 import datetime, time '''获取当前时间''' n =...

本文实例讲述了Python字典遍历操作。分享给大家供大家参考，具体如下： 1 遍历键值对可以使用一个 for 循环以及方法 items() 来遍历这个字典的键值对。 dict =...

宜配屋