Pytorch池化层Maxpool2d中ceil

    技术2022-07-11  110

    @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,舍弃
    Processed: 0.015, SQL: 10