How to visualize activation in neural network

2

If you just want to plot the activation maps, you could use this approach.