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

返回列表

文件对象提供了 read() 方法来按字节或字符读取文件内容，到底是读取宇节还是字符，则取决于是否使用了 b 模式，如果使用了 b 模式，则每次读取一个字节；如果没有使用 b 模式，则每...

本文实例讲述了Python实现的根据IP地址计算子网掩码位数功能。分享给大家供大家参考，具体如下： #!/usr/bin/env python # coding:utf-8 #!/b...

Python产生一个数值范围内的不重复的随机数，可以使用random模块中的random.sample函数，其用法如下： import random random.sample(po...

如下所示： #统计某文件夹下的所有csv文件的行数（多线程） import threading import csv import os class MyThreadLine(t...

预测结果转为numpy： logits=model(feature) #如果模型是跑在GPU上 result=logits.data.cpu().numpy() / logi...

宜配屋