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

返回列表

安装模块如果使用的是Linux系统，并且安装了pip，可以直接使用pip安装xlrd, xlwt： pip install xlwt pip install xlrd 也可以从官...

本文实例讲述了Python实现模拟登录及表单提交的方法。分享给大家供大家参考。具体实现方法如下： # -*- coding: utf-8 -*- import re import...

最近游戏项目在多个国家上线，每个国家都对应两份儿svn目录（一份是本地策划目录，一份是线上目录）。于是乎维护变得很烦躁。需要先更新本地策划svn目录，然后把更新的文件拷贝到对应的线上目录...

本文实例讲述了python删除特定文件的方法。分享给大家供大家参考。具体如下： #!/usr/bin/python # -*- coding: utf-8 -*- import os...

oop1.py文件代码 # user/bin/python class Foo: def bar(self): print('ok') def hello(self, name):...

宜配屋