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

返回列表

本文实例讲述了Python实现的插入排序算法原理与用法。分享给大家供大家参考，具体如下：插入排序的基本操作就是将一个数据插入到已经排好序的有序数据中，从而得到一个新的、个数加一的有序数...

最近将Pytorch程序迁移到GPU上去的一些工作和思考环境：Ubuntu 16.04.3 Python版本：3.5.2 Pytorch版本：0.4.0 0. 序言大家知道，在深度学...

本文实例讲述了python中字典(Dictionary)用法。分享给大家供大家参考。具体分析如下：字典（Dictionary）是一种映射结构的数据类型，由无序的“键－值对”组成。字典的...

本文实例讲述了Python统计纯文本文件中英文单词出现个数的方法。分享给大家供大家参考，具体如下：第一版: 效率低 # -*- coding:utf-8 -*- #!python3...

如何安装Python的操作步骤： 1.第一步先去python的官方网站下载python的安装包地址： https://www.python.org/downloads/ 根据自己的系...

宜配屋