[ Pytorch ] torch.squeeze() 和torch.unsqueeze()的用法

Posted 2021-03-12 recoverableti

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了[ Pytorch ] torch.squeeze() 和torch.unsqueeze()的用法相关的知识，希望对你有一定的参考价值。

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

squeeze()
torch.squeeze(a)：去掉a中维数为1的维度。
a.squeeze(N)：去掉特定维度N下维数为1的维度。
b=torch.squeeze(a，N)：a中去掉指定的维数为1的维度。

unsqueeze()
a.squeeze(N)：在a中指定位置N加上一个维数为1的维度。
b=torch.squeeze(a，N)：在a中指定位置N加上一个维数为1的维度。

以上是关于[ Pytorch ] torch.squeeze() 和torch.unsqueeze()的用法的主要内容，如果未能解决你的问题，请参考以下文章