Categories
Misc

Fully Vectorized Conv2D Implementation

Hey guys.

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. 🙂

submitted by /u/deepdipship
[visit reddit] [comments]

Leave a Reply

Your email address will not be published. Required fields are marked *