Hey, I have the activations of a conv layer with a single channel (normal shape pattern (batch, 1, rows, cols)) and would like to extract a window around the argmax of each activation in the batch. For example then, for patches of size 5x5 I'd like to get the shape (batch, 1, 5, 5). Is there any simple way to do this? I've tried some rather ugly workarounds with flattening and indices etc. but the gradients had taken ages to calculate making the entire thing un-trainable.
Thanks in advance -- --- You received this message because you are subscribed to the Google Groups "theano-users" group. To unsubscribe from this group and stop receiving emails from it, send an email to [email protected]. For more options, visit https://groups.google.com/d/optout.
