[ 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.unsqueeze(N) ：在a中指定位置N加上一個維數為1的維度。
b=torch.unsqueeze(a，N)：在a中指定位置N加上一個維數為1的維度。