I want to create a custom convolution operation by overriding, if possible, the torch.nn.functional.conv2d(). For example, I want to “add” the network weights instead of “multiplying” them, just for example.
I don’t want to write the forward propagation from scratch as that won’t allow me to use the backward(). In that case I’d have to calculate the gradients and all those stuffs by myself.