tensorflow之tf.squeeze()


tf.squeeze()函數的作用是從tensor中刪除所有大小(szie)是1的維度。

給定丈量輸入, 此操作返回的是相同類型的張量, 並刪除所有尺寸為1的維度。如果不想刪除所有尺寸為1的維度,

可以通過指定squeeze_dims來刪除特定維度。

下面通過例子來理解:

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]

shape(squeeze(t)) ==> [2, 3]   # 可見, 把shape為1的維度都刪除了。

 

或者刪除特定的維度:

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]

shape(squeeze(t, [2, 4])) ==> [1, 2, 3, 1]


免責聲明!

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



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