pytorch中squeeze()和unsqueeze()函数
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用法类似，结果相反
汇聚全球AI编程工具，助力开发者即刻编程。
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)