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,舍弃

你可能感兴趣的:(pytorch)