PyTorch中MaxPool的ceil_mode属性
参考博客:http://t.zoukankan.com/xxxxxxxxx-p-11529343.html
pytorch官方链接:https://pytorch.org/docs/stable/generated/torch.nn.MaxPool2d.html#torch.nn.MaxPool2d
PyTorch中MaxPool的ceil_mode属性
PyTorch中的MaxPool(最大池化)有一个属性:ceil_mode,默认为False(地板模式),为True时是天花板模式。
所以ceil模式就是会把不足square_size的边给保留下来,单独另算,或者也可以理解为在原来的数据上补充了值为-NAN的边。
而floor模式则是直接把不足square_size的边给舍弃了。