Custom Layers

http://en.diveintodeeplearning.org/chapter_deep-learning-computation/custom-layer.html