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的函數使用一樣 向上取整 或者向下取整
- 細節決定成敗