PyTorch中permute的用法详解


Posted in Python onDecember 30, 2019
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(1,3,2)可以把Tensor([[[1,2,3],[4,5,6]]]) 转换成

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

如果使用view(1,3,2),可以得到

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

以上这篇PyTorch中permute的用法详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Windows系统下使用flup搭建Nginx和Python环境的方法
Dec 25 Python
Python彩色化Linux的命令行终端界面的代码实例分享
Jul 02 Python
ubuntu系统下 python链接mysql数据库的方法
Jan 09 Python
Python中str.format()详解
Mar 12 Python
Python使用gRPC传输协议教程
Oct 16 Python
更改Python的pip install 默认安装依赖路径方法详解
Oct 27 Python
python实现动态创建类的方法分析
Jun 25 Python
python3+PyQt5 自定义窗口部件--使用窗口部件样式表的方法
Jun 26 Python
python画图的函数用法以及技巧
Jun 28 Python
tensorflow 限制显存大小的实现
Feb 03 Python
Python实现SMTP邮件发送
Jun 16 Python
python如何利用cv2模块读取显示保存图片
Jun 04 Python
python实现多进程按序号批量修改文件名的方法示例
Dec 30 #Python
Pytorch Tensor基本数学运算详解
Dec 30 #Python
python垃圾回收机制(GC)原理解析
Dec 30 #Python
利用Python代码实现一键抠背景功能
Dec 29 #Python
如何利用pygame实现简单的五子棋游戏
Dec 29 #Python
Python使用正则实现计算字符串算式
Dec 29 #Python
Django框架教程之中间件MiddleWare浅析
Dec 29 #Python
You might like
PHP insert语法详解
2008/06/07 PHP
PHP脚本中include文件出错解决方法
2008/11/20 PHP
thinkPHP3.x常量整理(预定义常量/路径常量/系统常量)
2016/05/20 PHP
Yii框架响应组件用法实例分析
2019/09/04 PHP
top.location.href 没有权限 解决方法
2008/08/05 Javascript
js控制web打印(局部打印)方法整理
2013/05/29 Javascript
详解JavaScript正则表达式中的global属性的使用
2015/06/16 Javascript
javascript控制台详解
2015/06/25 Javascript
js+html5实现可在手机上玩的拼图游戏
2015/07/17 Javascript
jQuery ajax调用后台aspx后台文件的两种常见方法(不是ashx)
2016/06/28 Javascript
Vue.js组件tabs实现选项卡切换效果
2016/12/01 Javascript
D3.js实现简洁实用的动态仪表盘的示例
2018/04/04 Javascript
浅谈Vue.use的使用
2018/08/29 Javascript
vue权限管理系统的实现代码
2019/01/17 Javascript
在vue项目中使用sass语法问题
2019/07/18 Javascript
微信小程序开发中var that =this的用法详解
2020/01/18 Javascript
微信小程序实现简单文字跑马灯
2020/05/26 Javascript
python函数返回多个值的示例方法
2013/12/04 Python
Python实现列表转换成字典数据结构的方法
2016/03/11 Python
matplotlib作图添加表格实例代码
2018/01/23 Python
python实现简单的单变量线性回归方法
2018/11/08 Python
Python numpy中矩阵的基本用法汇总
2019/02/12 Python
深入了解Django View(视图系统)
2019/07/23 Python
python3实现绘制二维点图
2019/12/04 Python
CSS3的新特性介绍
2008/10/31 HTML / CSS
html5实现canvas阴影效果示例
2014/05/07 HTML / CSS
HTML5 drag和drop具体使用详解
2021/01/18 HTML / CSS
股权收购意向书
2014/04/01 职场文书
消防标语大全
2014/06/07 职场文书
2014年个人债务授权委托书范本
2014/09/22 职场文书
领导干部作风整顿个人剖析材料
2014/10/11 职场文书
2014年师德师风工作总结
2014/11/25 职场文书
导游词欢迎词
2015/02/02 职场文书
天堂的孩子观后感
2015/06/11 职场文书
小学信息技术教学反思
2016/02/16 职场文书
24年收藏2000多部退役军用电台
2022/02/18 无线电