You could use forward hook and store some stats of the intermediate activations you are interested in. This post explains how to use forward hooks on modules.
You could use forward hook and store some stats of the intermediate activations you are interested in. This post explains how to use forward hooks on modules.