chickadee » nanograd » maxpool2d

maxpool2d input kernel-size #!key (stride #f)procedure

2-D max pooling with a square window of size kernel-size. The stride defaults to the kernel size, so windows do not overlap.

  • 3-D: (C, H, W) -> (C, OH, OW)
  • 4-D: (N, C, H, W) -> (N, C, OH, OW)

The gradient goes to the largest element of each window.