@Pytorch
当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,舍弃