Custom Conv2d Operation

vision
2

You could use unfold for your custom convolution (which might be slow) as described here.

1 Like