PyTorch中permute的使用方法


Posted in Python onApril 26, 2022

permute(dims)

将tensor的维度换位。

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

例:

import torch
import numpy    as np

a=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(多维数组,[维数的组合])

比如:

a=rand(2,3,4);  %这是一个三维数组,各维的长度分别为:2,3,4

%现在交换第一维和第二维:

permute(A,[2,1,3])  %变成3*2*4的矩阵

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

总结

到此这篇关于PyTorch中permute的基本用法的文章就介绍到这了!

Python 相关文章推荐
python计算书页码的统计数字问题实例
Sep 26 Python
Python计算程序运行时间的方法
Dec 13 Python
Python浅拷贝与深拷贝用法实例
May 09 Python
python批量制作雷达图的实现方法
Jul 26 Python
python os.listdir按文件存取时间顺序列出目录的实例
Oct 21 Python
基于树莓派的语音对话机器人
Jun 17 Python
pandas 数据结构之Series的使用方法
Jun 21 Python
Python的Tkinter点击按钮触发事件的例子
Jul 19 Python
python实现获取单向链表倒数第k个结点的值示例
Oct 24 Python
Flask缓存静态文件的具体方法
Aug 02 Python
python 逆向爬虫正确调用 JAR 加密逻辑
Jan 12 Python
Python使用sql语句对mysql数据库多条件模糊查询的思路详解
Apr 12 Python
Python matplotlib 利用随机函数生成变化图形
方法汇总:Python 安装第三方库常用
Apr 26 #Python
Python 统计序列中元素的出现频度
Apr 26 #Python
Python matplotlib安装以及实现简单曲线的绘制
Python爬虫 简单介绍一下Xpath及使用
分享python函数常见关键字
Apr 26 #Python
python和Appium的移动端多设备自动化测试框架
Apr 26 #Python
You might like
深入PHP变量存储的详解
2013/06/13 PHP
php操作mongodb封装类与用法实例
2018/09/01 PHP
PHP实现数组向任意位置插入,删除,替换数据操作示例
2019/04/05 PHP
用JQuery 判断某个属性是否存在hasAttr的解决方法
2013/04/26 Javascript
javascript强大的日期函数代码分享
2013/09/04 Javascript
jquery ajaxSubmit 异步提交的简单实现
2014/02/28 Javascript
兼容所有浏览器的js复制插件Zero使用介绍
2014/03/19 Javascript
JavaScript中Date对象的常用方法示例
2015/10/24 Javascript
JS实现的3D拖拽翻页效果代码
2015/10/31 Javascript
AngularJS 自定义指令详解及示例代码
2016/08/17 Javascript
JavaScript 闭包机制详解及实例代码
2016/10/10 Javascript
jQuery中的select操作详解
2016/11/29 Javascript
Javascript 实现全屏滚动实例代码
2016/12/31 Javascript
JavaScript之iterable_动力节点Java学院整理
2017/06/29 Javascript
vue2实现可复用的轮播图carousel组件详解
2017/11/27 Javascript
js点击时关闭该范围下拉菜单之外的菜单方法
2018/01/11 Javascript
一步快速解决微信小程序中textarea层级太高遮挡其他组件
2019/03/04 Javascript
详解小程序用户登录状态检查与更新实例
2019/05/15 Javascript
Vue proxyTable配置多个接口地址,解决跨域的问题
2020/09/11 Javascript
[02:12]2019完美世界全国高校联赛(春季赛)报名开启
2019/03/01 DOTA
[01:16:12]完美世界DOTA2联赛PWL S2 FTD vs Inki 第一场 11.21
2020/11/23 DOTA
解决Python 中英文混输格式对齐的问题
2018/07/16 Python
pytorch 实现将自己的图片数据处理成可以训练的图片类型
2020/01/08 Python
关于Kotlin中SAM转换的那些事
2020/09/15 Python
创造美妙香氛体验:Aera扩散器和香水
2018/11/25 全球购物
杭州时比特电子有限公司SQL
2013/08/22 面试题
应届生自我鉴定
2013/12/11 职场文书
财务总监管理职责范文
2014/03/09 职场文书
校园联欢晚会主持词
2014/03/17 职场文书
年度优秀员工获奖感言
2014/08/15 职场文书
社保缴纳证明申请书
2014/11/03 职场文书
廉洁自律个人总结
2015/02/14 职场文书
写给女朋友的保证书
2015/05/09 职场文书
致运动员的广播稿
2015/08/19 职场文书
Python jiaba库的使用详解
2021/11/23 Python
Matplotlib绘制条形图的方法你知道吗
2022/03/21 Python