1.这两个函数只能作用在列表或类列表的数据结构
2. squeeze()函数
该函数将某列表中,维度为1的维度删掉
如果是用squeeze(),则无论原列表有多少个1维的维度,全部删掉
squeeze()括号中可以添加维度的序号,以此删掉特定的维度
删掉了序号为2的那个维度
有意思的是,如果传入的序号,对应的维度不等于1,则squeeze()函数不起效,如下:
3.unsqueeze()函数
unsqueeze()函数就是添加一个一维的维度,具体添加在哪里跟传入的参数有关。
和squeeze()函数不同的是,unsqueeze()函数使用时必须加参数,不然会报错
括号里传入的参数就是新添的一维,在添加以后的总顺序里的序号。
但是,传入序号不能超过原最大序号+1,如下: