天天看點

pytorch中squeeze()和unsqueeze()函數介紹

一、unsqueeze()函數

1. 首先初始化一個a

pytorch中squeeze()和unsqueeze()函數介紹

可以看出a的次元為(2,3)

2. 在第二維增加一個次元,使其次元變為(2,1,3)

pytorch中squeeze()和unsqueeze()函數介紹

可以看出a的次元已經變為(2,1,3)了,同樣如果需要在倒數第二個次元上增加一個次元,那麼使用b.unsqueeze(-2)

二、squeeze()函數介紹

1. 首先得到一個次元為(1,2,3)的tensor(張量)

pytorch中squeeze()和unsqueeze()函數介紹

由圖中可以看出c的次元為(1,2,3)

2.下面使用squeeze()函數将第一維去掉

pytorch中squeeze()和unsqueeze()函數介紹

可見,次元已經變為(2,3)

3.另外

pytorch中squeeze()和unsqueeze()函數介紹

可以看出次元并沒有變化,仍然為(1,2,3),這是因為隻有次元為1時才會去掉。

一、unsqueeze()函數