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

Posted 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()的用法的主要内容,如果未能解决你的问题,请参考以下文章