permute（）--对维度进行换位

permute(dims)

将tensor的维度换位。

参数：参数是一系列的整数，代表原来张量的维度。比如三维就有0，1，2这些dimension。

例：

import torch
import numpy    as npa=np.array([[[1,2,3],[4,5,6]]])unpermuted=torch.tensor(a)
print(unpermuted.size())  #  ——>  torch.Size([1, 2, 3])permuted=unpermuted.permute(2,0,1)
print(permuted.size())     #  ——>  torch.Size([3, 1, 2])

比如图片img的size比如是（28，28，3）就可以利用img.permute(2,0,1)得到一个size为（3，28，28）的tensor。

利用这个函数permute（0，2，1）可以把Tensor([[[1,2,3],[4,5,6]]]) 转换成

tensor([[[1., 4.],[2., 5.],[3., 6.]]])

如果使用view，可以得到

tensor([[[1., 2.],[3., 4.],[5., 6.]]])

permute（）--对维度进行换位相关推荐

Pytorch中tensor.view().permute().contiguous()函数理解
Pytorch中tensor.view().permute().contiguous()函数理解 yolov3中有一行这样的代码,在此记录一下三个函数的含义 # 例子中batch_size为整型,le ...
Pytorch 笔记——tensor
tensor和numpy中的ndarray很相似,唯一的不同是tensor可以应用在GPU上,来加速计算. 1 创建tensor randn(size) 创建一个标准正态分布内随机的矩阵 zeros( ...
利用TCN网络实现MNIST手写体数据集的识别
利用TCN网络实现MNIST手写体数据集的识别 TCN识别MNIST的GitHub网址 https://github.com/locuslab/TCN 论文来源 https://arxiv.org/p ...
目标检测(四)：SSD之Pytorch源码解读
读完 SSD 的论文内容能大致了解这一算法的核心思想和算法流程,但要将其应用到实际问题上还需要去读代码.论文给出的 SSD 源码是用 Caffe 框架实现的,但自己使用 Caffe 搭建 SSD 的环 ...
对视觉显著性检测（Saliency Object Detection）中Channel Attention的一些总结
因为最近的paper涉及,所以对channel attention进行了了解,比较浅显,但是以此篇博客记录,之后有相关进展,也会更新.已经有很多相关的文章对Attention的本质,原理进行介绍.例如 ...
自动驾驶知识点 Review 2D 感知算法一（两阶段法 R-CNN系列，FPN，R-FCN）
文章目录 R-CNN系列 R-CNN R-CNN Fast RCNN Faster RCNN FPN R-FCN R-CNN系列 R-CNN 两个步骤:1)提取物体区域:2)对区域进行分类识别: 本章 ...
第六课.NLP文本分类任务
第六课目录 NLP文本分类简介 IMDB数据集准备设置随机种子下载IMDB并划分数据集构建词汇表使用torchtext生成batch WordAveraging 模型定义加载由glove.6 ...
pytorch-词嵌入基础
词嵌入基础我们在"循环神经网络的从零开始实现"一节中使用 one-hot 向量表示单词,虽然它们构造起来很容易,但通常并不是一个好选择.一个主要的原因是,one-hot 词向量无 ...
PyTorch 11—简单图像定位
常见图像处理的任务 (1)分类给定一幅图像,我们用计算机模型预测图片中有什么对象 . (2)分类加定位不仅需要我们知道图片中的对象是什么,还要在对象的附近画一个边框,确定该对象所处的位置. (3) ...

permute（）--对维度进行换位

permute(dims)

permute（）--对维度进行换位相关推荐

最新文章

热门文章