自定义Keras层必须继承tf.keras.layers.Layer，显式调用super().__init__(**kwargs)，在build()中用add_weight()创建可训练参数，在call()中执行计算；多输入/输出需重写call和compute_output_shape；保存加载需通过custom_objects注册或使用SavedModel格式。
自定义Keras层必须继承tf.keras.layers.Layer
不继承这个基类，就无法被Keras模型识别为合法层——哪怕你写了call()方法，也会在构建模型时报TypeError: Layer is not a valid layer。Keras依赖Layer的内部协议（如build()触发、权重注册、前向传播调度），跳过它等于绕开整个机制。
常见错误是直接写一个普通类 + __call__()，或者误继承tf.keras.Model——后者是用于封装多层逻辑的容器，不是单层实现方式。
- 必须显式调用
super().__init__(**kwargs)，否则self.name、self.trainable等基础属性不会初始化 - 如果需要可训练参数，必须在
build(self, input_shape)中创建，并用self.add_weight()注册；不能在__init__里直接用tf.Variable -
input_shape是TensorShape对象（如(None, 128)），注意索引时用input_shape[-1]取特征维，而非input_shape[1]（batch维可能为None）
build()和call()分工要清晰
build()只负责“定义参数”，不执行计算；call()只负责“执行计算”，不创建新变量。混用会导致两次构建（如模型重复fit()）、变量重复注册或ValueError: Variable already exists。
典型场景：你想实现一个带缩放的ReLU（alpha * relu(x)），其中alpha是可学习标量：
立即学习“Python免费学习笔记（深入）”；
class ScalableReLU(tf.keras.layers.Layer):
def __init__(self, **kwargs):
super().__init__(**kwargs)
<pre class="brush:php;toolbar:false;">def build(self, input_shape):
# ✅ 正确：在build里声明权重
self.alpha = self.add_weight(
shape=(), initializer='ones', trainable=True, name='alpha'
)
def call(self, inputs):
# ✅ 正确：在call里用已有权重运算
return self.alpha * tf.nn.relu(inputs)
如果把self.alpha = tf.Variable(...)写在__init__里，Keras不会将其纳入权重管理，model.trainable_weights里看不到它；如果在call()里调用self.add_weight()，每次前向都会尝试新建变量，立刻报错。
处理动态batch size和多输入/输出需显式声明
Keras默认假设输入是(batch, ...)格式，但如果你的层要支持None batch维（即任意batch size），build()收到的input_shape第一个维度就是None。此时别用input_shape[0]做形状推导——它会是None，应从第二维起算。
多输入层（如接收[x, mask]）必须重写call(self, inputs)并接受list或dict，且需在compute_output_shape()中手动返回对应形状（否则model.summary()会出错或显示multiple）：
- 多输入：参数
inputs是list，用inputs[0]、inputs[1]分别取；不要试图解包成def call(self, x, mask) - 多输出：返回
tuple（如return out1, out2），同时重写compute_output_shape返回对应tuple形状 - 若输出形状依赖输入值（如mask长度），必须在
call()里用tf.shape(inputs)[0]动态获取，不能只靠input_shape
保存与加载自定义层必须注册或传入custom_objects
用model.save('my_model.h5')或tf.keras.models.save_model()保存含自定义层的模型后，加载时会报ValueError: Unknown layer: ScalableReLU——因为Keras序列化时只存类名，不存代码。
两种解决方式：
- 加载时显式传
custom_objects={'ScalableReLU': ScalableReLU}：tf.keras.models.load_model('my_model.h5', custom_objects={...}) - 用
tf.keras.utils.get_custom_objects().update({...})全局注册（适合调试，生产环境慎用） - 更稳妥的做法：改用
save_format='tf'（即SavedModel格式），它会序列化call()的tf.function图，但前提是你的层不依赖外部Python状态（如随机数生成器、文件句柄）
最容易被忽略的是：自定义层里的add_weight()如果用了lambda初始化器（如lambda: tf.random.normal(())），保存加载后该权重会重初始化——必须用字符串名（如'random_normal'）或tf.keras.initializers实例。
