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