pytorch增加一维_pytorch 数据维度变换

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

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值