大家好,我是爱编程的喵喵。双985硕士毕业,现担任全栈工程师一职,热衷于将数据思维应用到工作与生活中。从事机器学习以及相关的前后端开发工作。曾在阿里云、科大讯飞、CCF等比赛获得多次Top名次。现为CSDN博客专家、人工智能领域优质创作者。喜欢通过博客创作的方式对所学的知识进行总结与归纳,不仅形成深入且独到的理解,而且能够帮助新手快速入门。
本文主要介绍了tensorflow2/keras中如何自定义layer,在经过了亲身的实践后,终于找到了可复现的实战代码,最终将详细的代码总结如下,希望能对学习tensorflow2/keras的同学有所帮助。
文章目录
- 1. 自定义layer简单实例
- 2. tf.keras.layers.Layer源码解析
- 3. 总结及案例
1. 自定义layer简单实例
虽然tf.keras.layers为我们封装了一些高级API(已经定义好了一些现成的layer,详情见https://catcoder.blog.csdn.net/article/details/106657923),但有时并不能满足我们的需求,因此需要构建自己的layer,下面