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连接sqlserver读取数据库，读取数据时候在本机电脑正常，但是把程序部署到服务器运行时一直报错“未发现数据源名称并且未指定默认驱动程序”，后来发现是因为数据源的问...

利用Python中的randomw.sample()函数实现 resultList=random.sample(range(A,B),N); #表示从[A,B]间随机生成N个数，结...

本文实例为大家分享了python实现名片管理系统源代码，供大家参考，具体内容如下 import os def print_menu(): print("*"*50) print(...

在很多神经网络中，往往会出现多个层共享一个权重的情况，pytorch可以快速地处理权重共享问题。例子1： class ConvNet(nn.Module): def __ini...

第一步接下来我们将看见如何在 Python 中运行一个传统的“Hello World”程序。Python教程本章将会教你如何编写、保存与运行 Python 程序。通过 Python...

宜配屋