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.