一、unsqueeze()函數
1. 首先初始化一個a
可以看出a的次元為(2,3)
2. 在第二維增加一個次元,使其次元變為(2,1,3)
可以看出a的次元已經變為(2,1,3)了,同樣如果需要在倒數第二個次元上增加一個次元,那麼使用b.unsqueeze(-2)
二、squeeze()函數介紹
1. 首先得到一個次元為(1,2,3)的tensor(張量)
由圖中可以看出c的次元為(1,2,3)
2.下面使用squeeze()函數将第一維去掉
可見,次元已經變為(2,3)
3.另外
可以看出次元并沒有變化,仍然為(1,2,3),這是因為隻有次元為1時才會去掉。
一、unsqueeze()函數