Pytorch池化层Maxpool2d中ceil_mode参数

@Pytorch

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

你可能感兴趣的:(python,Pytorch,Python,Deep,Learnin,Marchine,Lea)