Only works for networks with dense layers

[sobhanhemati]

Thanks for the implementation.

Have you tried this with CNN? For me, it gives an error. I

I guess there is a bug in code when you get layer shapes in the " get_vectors" function.

[songofphysics]

This is a non-issue. The code is clearly meant for a Fully Connected Neural Network. A CNN is not fully connected.