I wrote a post describing in detail a full vectorized implementation of the convolution operation in NumPy: https://lucasdavid.github.io/vectorization/
I would appreciate if you could give me any notes. I’m also trying to translate this to TensorFlow, but it’s not as trivial as I initially thought, considering indexing is very different (my implementation relies on selecting the multiple regions at once with `image[…, r, c]`, where `r` and `c` are two index matrices).
Any ideas on this would be greatly appreciated!
Have a great day. 🙂