pytorch中squeeze()和unsqueeze()函数

squeeze的用法主要就是对数据的维度进行压缩。

unsqueeze的用法主要就是对数据的维度进行解压。

squeeze

1）a.squeeze()就是讲 a中所有为1的维度删掉

 

2）squeeze(a)等同于a.squeeze()

3） torch.squeeze(a，N) ，如果a的第N维，维度为1，则去掉；不为1则方法失效

4）a.squeeze(1)效果同上

 

unsqueeze

 

1)  与上述squeeze用法类似，结果相反