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


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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM