squeeze的用法主要就是對數(shù)據(jù)的維度進(jìn)行壓縮或者解壓。

先看torch.squeeze() 這個(gè)函數(shù)主要對數(shù)據(jù)的維度進(jìn)行壓縮，去掉維數(shù)為1的的維度，比如是一行或者一列這種，一個(gè)一行三列（1,3）的數(shù)去掉第一個(gè)維數(shù)為一的維度之后就變成（3）行。squeeze(a)就是將a中所有為1的維度刪掉。不為1的維度沒有影響。a.squeeze(N) 就是去掉a中指定的維數(shù)為一的維度。還有一種形式就是b=torch.squeeze(a，N) a中去掉指定的定的維數(shù)為一的維度。

再看torch.unsqueeze()這個(gè)函數(shù)主要是對數(shù)據(jù)維度進(jìn)行擴(kuò)充。給指定位置加上維數(shù)為一的維度，比如原本有個(gè)三行的數(shù)據(jù)（3），在0的位置加了一維就變成一行三列（1,3）。a.squeeze(N) 就是在a中指定位置N加上一個(gè)維數(shù)為1的維度。還有一種形式就是b=torch.squeeze(a，N) a就是在a中指定位置N加上一個(gè)維數(shù)為1的維度

參考資料：https://blog.csdn.net/hustchenze/article/details/78989426
https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.squeeze.html

總結(jié)

以上是生活随笔為你收集整理的pytorch学习 中 torch.squeeze() 和torch.unsqueeze()的用法的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。