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 28 Python
python如何实现excel数据添加到mongodb
Jul 30 Python
Tornado协程在python2.7如何返回值(实现方法)
Jun 22 Python
Python闭包之返回函数的函数用法示例
Jan 27 Python
Odoo中如何生成唯一不重复的序列号详解
Feb 10 Python
CentOS7下python3.7.0安装教程
Jul 30 Python
python三大神器之fabric使用教程
Jun 10 Python
Pytorch 实现冻结指定卷积层的参数
Jan 06 Python
关于初始种子自动选取的区域生长实例(python+opencv)
Jan 16 Python
Python ini文件常用操作方法解析
Apr 26 Python
pandas.DataFrame.drop_duplicates 用法介绍
Jul 06 Python
Jupyter Notebook 如何修改字体和大小以及更改字体样式
Jun 03 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
windows xp下安装pear
2006/12/02 PHP
php object转数组示例
2014/01/15 PHP
javascript web对话框与弹出窗口
2009/02/22 Javascript
javascript Array对象基础知识小结
2010/11/16 Javascript
Node.js与PHP、Python的字符处理性能对比
2014/07/06 Javascript
解决jquery实现的radio重新选中的问题
2015/07/03 Javascript
js中flexible.js实现淘宝弹性布局方案
2020/06/23 Javascript
jQuery遍历json的方法(推荐)
2016/06/12 Javascript
浅析JS获取url中的参数实例代码
2016/06/14 Javascript
JavaScript中push(),join() 函数 实例详解
2016/09/06 Javascript
详解nodejs 文本操作模块-fs模块(二)
2016/12/22 NodeJs
jQuery用noConflict代替$的实现方法
2017/04/12 jQuery
Vue计算属性的使用
2017/08/04 Javascript
react native仿微信PopupWindow效果的实例代码
2017/08/07 Javascript
记录一次开发微信网页分享的步骤
2019/05/07 Javascript
2019年度web前端面试题总结(主要为Vue面试题)
2020/01/12 Javascript
解决vuex刷新数据消失问题
2020/11/12 Javascript
详解javascript脚本何时会被执行
2021/02/05 Javascript
Python中使用tarfile压缩、解压tar归档文件示例
2015/04/05 Python
python3音乐播放器简单实现代码
2020/04/20 Python
Python基于jieba库进行简单分词及词云功能实现方法
2018/06/16 Python
Python Scapy随心所欲研究TCP协议栈
2018/11/20 Python
python3 enum模块的应用实例详解
2019/08/12 Python
Tensorflow Summary用法学习笔记
2020/01/10 Python
通过python检测字符串的字母
2020/02/18 Python
Python-jenkins 获取job构建信息方式
2020/05/12 Python
如何利用python web框架做文件流下载的实现示例
2020/06/02 Python
中国电子产品外贸网站:MiniIntheBox
2017/02/06 全球购物
英国Iceland杂货店:网上食品购物
2020/12/16 全球购物
仓库主管的岗位职责
2013/12/04 职场文书
自动化专业职业生涯规划书范文
2014/01/16 职场文书
宾馆总经理岗位职责
2014/02/14 职场文书
房地产销售员岗位职责
2015/04/11 职场文书
大学团日活动总结书
2015/05/11 职场文书
2016小学教师读书心得体会
2016/01/13 职场文书
护理自荐信
2019/05/14 职场文书