自定义层

https://zh-v2.d2l.ai/chapter_deep-learning-computation/custom-layer.html