Describe the feature and motivation

It would be great to have a max/min pooling operation similar to pytorch's MaxPool layer. I'd imagine max and min pooling function that takes a kernel size, stride and operates on arbitrary number of channels.

Additional context

I specifically use this to extract foreground/background in a grayscale image. Would be great to have every sub function in one image processing library instead of using torch/scikit-image (I use openCV in python)