I’m trying to write a “Conv2DFunction” in the same spirit as the class LinearFunction(Function) here: https://pytorch.org/docs/master/notes/extending.html 4
But, I can’t quite wrap my head around how to calculate gradients in the backward method, because with convolution kernels you can’t use simple matmuls to calculate gradients. Any hints or help is appreciated.
Thanks!