How to get output of layers?

You could use forward hook to get the layers outputs as described here.

2 Likes