Thanks, that’s very helpful.
My case is basically the following: I want to obtain max pooling indices from a tensor A, and then get the values at the locations of those indices in a tensor B of the same dimensionality. So far, I haven’t figured how to do this and was thinking I could do this by using the fwd/bwd implementations of max_pooling2d, but maybe I’m making things to complicated?