I am implementing a convolutional layer with customized convolution operator (such as deformable convolution). I find that I have to extend Pytorch by C and now I want to find the source code of convolution. Or is there any other simpler ways to do it?
I would like to implement a custom convolution layer, where in I actually modify each individual part of the input with other external filter which also requires gradient.
I am succesful in implementing that in python, but it is very slow as expected.