Yes, you can use forward hooks as described here to inspect and manipulate the output activations.
1 Like
Yes, you can use forward hooks as described here to inspect and manipulate the output activations.