自定义层

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