一、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時才會去掉。