博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
pytorch增加一维_pytorch 数据维度变换
阅读量:5898 次
发布时间:2019-06-19

本文共 754 字,大约阅读时间需要 2 分钟。

1、view和reshape:

* 两者功能完全一样。只是pytorch 0.3版本默认是view,为了和Numpy一致,后来增加了reshape的api

* 注意:变换前后的数据大小必须一样

20200328104122553031.png

2、squeeze v.s. unsqueeze

1)unsqueeze,将维度展开。

20200328104122822545.png

如下图所示:

一维数据插入-1以后,将数据变成二维(1行2列,变为2行1列)

如果插入0,会将1行2列,变成1行1列(1列里面又是1行1列)

20200328104122966090.png

2)squeeze:删减维度

与unsqueeze作用刚好相反。

a.squeeze()默认是删减所有为1的维度数据,如下图所示:

20200328104123033469.png

3、expand v.s. repeat

1)expand:扩展维度到指定维度

20200328104123156508.png

2)repeat:

复制原维度多少倍。(不建议使用,因为复制数据会重新申请内存空间,比较占内存)

20200328104123275641.png

4、t:矩阵转置:

20200328104123467035.png

5、transpose v.s. permute

1)transpose:维度交换

20200328104123563708.png

torch.all、torch.eq  # 用于比较两个数据是否一致

2)permute:

调用一次或多次的transpose来达到交换

20200328104123663311.png

6、broadcast广播(自动扩展):

1)特点:

自动扩展维度,并且不用复制数据,节省空间

20200328104124737461.png

2)不能broadcast的情况:

给定A的维度[4 32 8],若给定B[4]。因为广播(默认)是从小维度开始,即此处的8所在维度。因为4属于8的一半维度,而另外的两个维度可以扩展为4 32,最终导致8所在维度缺少一半数据,所以此时不能扩张。(如果给定[1],可以8的每个数字加上1,或者给定[8],每个元素一一对应相加,否则不能扩张)

3)广播例子:

20200328104124882960.png

20200328104124965962.png

4)综上,可以广播的条件:

要么不给定维度,要么给定1,要么给定原数据相同的维度。

20200328104125065565.png

转载地址:http://xrqsx.baihongyu.com/

你可能感兴趣的文章
Linux学习之CentOS(八)--Linux系统的分区概念
查看>>
System.Func<>与System.Action<>
查看>>
asp.net开源CMS推荐
查看>>
csharp skype send message in winform
查看>>
MMORPG 游戏服务器端设计--转载
查看>>
SILK 的 Tilt的意思
查看>>
Html学习笔记3
查看>>
HDFS dfsclient写文件过程 源码分析
查看>>
ubuntu下安装libxml2
查看>>
nginx_lua_waf安装测试
查看>>
WinForm窗体缩放动画
查看>>
JQuery入门(2)
查看>>
linux文件描述符
查看>>
传值引用和调用引用的区别
查看>>
hyper-v 无线网连接
查看>>
Python3.7.1学习(六)RabbitMQ在Windows环境下的安装
查看>>
Windows下memcached的安装配置
查看>>
ubuntu: firefox+flashplay
查看>>
常见的海量数据处理方法
查看>>
web.xml 中CharacterEncodingFilter类的学习
查看>>