Pytorch池化层Maxpool2d中ceil_mode参数

当ceil_mode = true时,将不足kernel_size大小的数据保存,自动补足NAN至kernel_size大小;

当ceil_mode = False时,剩余数据不足kernel_size大小时,直接舍弃。

例如:

inputs的大小为5*5, max_pool中kernel_size为2

inputs= [0 0 0 0 0
              1 1 1 1 1
              2 2 2 2 2
              3 3 3 3 3
              4 4 4 4 4]

 当ceil_mode = True时:

              0   0 | 0   0 | 0 ×
              1   1 | 1   1 | 1 ×
       ————————————
              2   2 | 2   2 | 2 ×
              3   3 | 3   3 | 3 ×
       ————————————
              4   4 | 4   4 | 4 ×
              × × | ×  ×| ××

输出:
              [1 1 1
               3 3 3
               4 4 4]

即:数据不足以构成2*2,仍然对剩余数据进行计算

当ceil_mode = False时:
              0   0 | 0   0
              1   1 | 1   1
        ————————
              2   2 |  2   2
              3   3 |  3   3
输出:

              [1   1
               3  3]
即:数据不足以构成2*2,舍弃

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值