Hi
I have a very specific operation I’d like to implement in PyTorch, it has no trainable parameters but needs to be differentiable. The operation works as follows:

Take a fixed size window eg. 7x7
Take the mean of the left half of the window and the mean of the right half of the window (always ignoring the center column)
Set the center value to the ratio of the left side to the right side ( output(3,3) = MEANleft / MEANright)
Repeat over the entire image (No padding needed)

I have implemented this forward method using loops but it is really slow. Is there a simple way to do this with Tensor Comprehensions? Or a better way to apply such an operation using existing layers?
Thanks