TensorFlow2.0学习笔记-4.模型的自定义

4.模型的自定义 4.1.自定义层 使用的主要数据结构是Layer 实现自定义层的最佳方法是扩展tf.keras.layers.Layer类并实现： • __init__ ：可以在其中进行所有与输入无关的初始化，定义相关的层 • build： 知道输入张量的形状并可以进行其余的初始化 • call： 在这里进行前向传播 注意：不一定需要在build中创建变量时，也可以在__init__中创建它们。

>>阅读原文<<