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用法类似,结果相反
squeeze的用法主要就是对数据的维度进行压缩。
unsqueeze的用法主要就是对数据的维度进行解压。
1)a.squeeze()就是讲 a中所有为1的维度删掉
2)squeeze(a)等同于a.squeeze()
3) torch.squeeze(a,N) ,如果a的第N维,维度为1,则去掉;不为1则方法失效
4)a.squeeze(1)效果同上
1) 与上述squeeze用法类似,结果相反