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 相关文章推荐
Python自动化测试工具Splinter简介和使用实例
May 13 Python
详解python单例模式与metaclass
Jan 15 Python
PyCharm使用教程之搭建Python开发环境
Jun 07 Python
python3实现多线程聊天室
Dec 12 Python
python利用多种方式来统计词频(单词个数)
May 27 Python
python 搜索大文件的实例代码
Jul 08 Python
pandas DataFrame的修改方法(值、列、索引)
Aug 02 Python
Pytorch 多维数组运算过程的索引处理方式
Dec 27 Python
Python3操作读写CSV文件使用包过程解析
Apr 10 Python
使用python脚本自动生成K8S-YAML的方法示例
Jul 12 Python
Python Unittest原理及基本使用方法
Nov 06 Python
Django集成MongoDB实现过程解析
Dec 01 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中在PDO中使用事务(Transaction)
2011/05/14 PHP
php 调试利器debug_print_backtrace()
2012/07/23 PHP
PHP中VC6、VC9、TS、NTS版本的区别与用法详解
2013/10/26 PHP
laravel 创建命令行命令的图文教程
2019/10/23 PHP
tp5框架前台无限极导航菜单类实现方法分析
2020/03/29 PHP
iphone safari不支持position fixed的解决方法
2012/05/04 Javascript
jquery 插件学习(二)
2012/08/06 Javascript
JavaScript返回网页中锚点数目的方法
2015/04/03 Javascript
Jquery实现瀑布流布局(备有详细注释)
2015/07/31 Javascript
JS实现横向拉伸动感伸缩菜单效果代码
2015/09/04 Javascript
Kindeditor在线文本编辑器如何过滤HTML
2016/04/14 Javascript
jQuery事件的绑定、触发、及监听方法简单说明
2016/05/10 Javascript
Bootstrap组件系列之福利篇几款好用的组件(推荐)
2016/06/23 Javascript
js实现常用排序算法
2016/08/09 Javascript
React Native实现简单的登录功能(推荐)
2016/09/19 Javascript
node.js 抓取代理ip实例代码
2017/04/30 Javascript
jQuery.validate.js表单验证插件的使用代码详解
2018/10/22 jQuery
使用vue-router切换页面时实现设置过渡动画
2019/10/31 Javascript
vuex实现数据状态持久化
2019/11/11 Javascript
JavaScript实现简单贪吃蛇效果
2020/03/09 Javascript
使用纯前端JavaScript实现Excel导入导出方法过程详解
2020/08/07 Javascript
Python中处理unchecked未捕获异常实例
2015/01/17 Python
Python中用psycopg2模块操作PostgreSQL方法
2017/11/28 Python
python 爬虫 批量获取代理ip的实例代码
2018/05/22 Python
python中利用h5py模块读取h5文件中的主键方法
2018/06/05 Python
Numpy之文件存取的示例代码
2018/08/03 Python
django框架事务处理小结【ORM 事务及raw sql,customize sql 事务处理】
2019/06/27 Python
Python datetime模块使用方法小结
2020/06/18 Python
tensorflow 2.1.0 安装与实战教程(CASIA FACE v5)
2020/06/30 Python
Django admin组件的使用
2020/10/24 Python
Pytorch自定义Dataset和DataLoader去除不存在和空数据的操作
2021/03/03 Python
编辑硕士自荐信范文
2013/11/27 职场文书
毕业证委托书范文
2014/09/26 职场文书
2014年学生资助工作总结
2014/12/18 职场文书
中国汉字听写大会观后感
2015/06/02 职场文书
单身狗福利?Python爬取某婚恋网征婚数据
2021/06/03 Python