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脚本为敏捷开发脚本，在zabbix监控也起到重要作用，以下是使用python脚本发送告警邮件配置方法。脚本如下： #!/usr/bin/python #coding:u...

Python使用QRCode模块生成二维码 QRCode官网 https://pypi.python.org/pypi/qrcode/5.1 简介 python-qrcode是个用来...

Python获取电脑截图有多种方式，具体如下： PIL中的ImageGrab模块 windows API PyQt pyautogui PIL中的ImageGrab模块 impor...

需求分析 1、读取指定目录下的所有文件 2、读取指定文件，输出文件内容 3、创建一个文件并保存到指定目录实现过程 Python写代码简洁高效，实现以上功能仅用了40行左右...

本文实例讲述了python实现在sqlite动态创建表的方法。分享给大家供大家参考。具体实现方法如下： import sqlite3 as db conn = db.connect(...

宜配屋