seungribariumgd的博客

原创 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]