Custom Layers

http://d2l.ai/chapter_deep-learning-computation/custom-layer.html