[转载]PyTorch中permute的用法
2024-08-27 19:26:51
[转载]PyTorch中permute的用法
来源:https://blog.csdn.net/york1996/article/details/81876886
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.]]])
注意,上面例子中使用view是直接在原tensor上操作的,而不是在permute之后的结果上
最新文章
- 使用 HTML5 Canvas 绘制出惊艳的水滴效果
- zabbix3.0安装教程
- JSF 嵌套
- PHP面向对象(OOP)编程入门教程————如何实例化对象?
- away3d学习
- 从程序员到CTO的Java技术路线图 (转自安卓巴士)
- 8套迷人精致的CSS3 3D按钮动画
- 如何删除google流氓扩展(强制安装,并且无权限删除)
- 初识nginx+tomcat
- [转]给Linux系统管理员准备的Nmap命令的29个实用范例+ tsysv 系统服务器管理器
- android网络编程之HttpUrlConnection的讲解--实现文件的断点上传
- 传统平面广告已OUT出局,VR全景异军突起——VR全景智慧城市
- Unity LayerMask
- python基础-小练习
- ActiveReports 大数据分析报告:2019软件开发者现状
- Centos7 kernel 内核升级 GPU显卡驱动程序编译安装
- c#两个listbox怎么把内容添加到另外个listbox
- container(容器),injection(注入)
- window 10 专业版激活|win 10专业版激活码
- VS Code Html Zen coding
热门文章
- python笔记2小数据池,深浅copy,文件操作及函数初级
- 阶段5 3.微服务项目【学成在线】_day03 CMS页面管理开发_03-自定义查询页面-前端
- php文件断点上传
- LeetCode_9. Palindrome Number
- Vue学习笔记(三)组件间如何通信传递参数
- QFramework 使用指南 2020 (四):脚本生成(2)ViewController 与 ViewController 嵌套绑定
- DB2分页
- RestHighLevelClient 之 Scroll
- oauth2中org.springframework.security.core.userdetails.User无法转换为封装的AuthorizationInfoBean
- MIT 6.828 课程介绍