squeeze()和unsqueeze()函数辨析

1.这两个函数只能作用在列表或类列表的数据结构

在这里插入图片描述
在这里插入图片描述

2. squeeze()函数

该函数将某列表中,维度为1的维度删掉
如果是用squeeze(),则无论原列表有多少个1维的维度,全部删掉
在这里插入图片描述
squeeze()括号中可以添加维度的序号,以此删掉特定的维度
在这里插入图片描述
删掉了序号为2的那个维度
有意思的是,如果传入的序号,对应的维度不等于1,则squeeze()函数不起效,如下:
在这里插入图片描述

3.unsqueeze()函数

unsqueeze()函数就是添加一个一维的维度,具体添加在哪里跟传入的参数有关。
和squeeze()函数不同的是,unsqueeze()函数使用时必须加参数,不然会报错
在这里插入图片描述
括号里传入的参数就是新添的一维,在添加以后的总顺序里的序号。
但是,传入序号不能超过原最大序号+1,如下:
在这里插入图片描述


版权声明:本文为frank_haha原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。