seungribariumgd的博客

原创 Pytorch池化层Maxpool2d中ceil_mode参数

@PytorchPytorch迟化层Maxpool2d中ceil_mode参数当ceil_mode = true时，将保存不足为kernel_size大小的数据保存，自动补足NAN至kernel_size大小；当ceil_mode = False时，剩余数据不足kernel_size大小时，直接舍弃。例如：inputs的大小为5*5， max_pool中kernel_size为2inputs= [0 0 0 0 01 1 1 1 12 2 2 2 23 3 3 3 34 4 4 4 4]