hi
I recently implemented a convolution with the PyTorch extension here 1 and the code is based on here 1.
It is worth mentioning that the convolution achieved by PyTorch is very inefficient:

3_11.jpg1302×716 182 KB

I hope this helps you !