Heatmap localization

To get the intermediate activation output of a layer you can use forward hooks as described by @carloalbertobarbano. This post gives an example how to use them.