PyTorch ------MaxPool池化ceil_mode使用

Pytorch _MaxPoolNd類

  • 這個類見的少,但是MaxPool1d、MaxPool2d、MaxPool3d應該很常見了.
  • 在源碼中MaxPool1d、MaxPool2d、MaxPool3d 這三類都是繼承 _MaxPoolNd這個基類的
  • 在這裏插入圖片描述
  • 在這裏插入圖片描述
  • 在這裏插入圖片描述
  • 源碼截圖
  • 基類_MaxPoolNd中存在一個屬性ceil_mode
  • 在這裏插入圖片描述
  • 這個屬性在Input size 爲偶數的時候,沒有影響,但是當Input size 爲奇數時output size 就不一樣了,ceil_model = True 和false 是不同的結果
  • 在這裏插入圖片描述
  • 下面用代碼來展示其結果
  • ceil_mode = False
  • ⚠️⚠️⚠️⚠️⚠️⚠️ 輸出結果
  • 在這裏插入圖片描述
  • ceil_mode = True
  • ⚠️⚠️⚠️⚠️⚠️⚠️ 輸出結果
  • 在這裏插入圖片描述
  • ceil_mode 類似math庫中的ceil和floor的函數使用一樣 向上取整 或者向下取整
  • 細節決定成敗
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章