pytorch permute维度转换方法

yipeiwu_com6年前 (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设计】。

返回列表

本文实例讲述了Python使用numpy产生正态分布随机数的向量或矩阵操作。分享给大家供大家参考，具体如下：简单来说，正态分布（Normal distribution）又名高斯分布（G...

由于psutil已更新到3.0.1版本，最新的代码如下： #!/usr/bin/env python import os import time import sys import...

索引和切片是NumPy中最重要最常用的操作。熟练使用NumPy切片操作是数据处理和机器学习的前提，所以一定要掌握好。文档：https://docs.scipy.org/doc/num...

绘制八个子图 import matplotlib.pyplot as plt fig = plt.figure() shape=['.','o','v','>','<...

Python下一切皆对象，每个对象都有多个属性(attribute)，Python对属性有一套统一的管理方案。 __dict__与dir()的区别： dir()是一个函数，返回的是lis...

宜配屋