Visualize feature maps of the 1st layer

[ptrblck]

x is just a random input which is used as an example to show how hooks are used.