如何为Keras自定义Layer中的子Dense层权重命名？

这个问题的核心原因是你在父层初始化完成前就创建了子层，导致Keras无法为子层分配正确的命名空间。解决方法很简单，只需要调整初始化顺序，或者显式注册子层，就能让子层的name参数生效，权重名称会自动带上子层标识。

修正方案一：先初始化父层，再创建子层 #

在自定义Layer的__init__方法中，先调用super().__init__(**kwargs)完成父层初始化，再创建子层。这样Keras会自动识别这些子层，并为它们在父层的命名空间下创建独立名称：

import tensorflow as tf
from tensorflow import keras

class DoubleDense(keras.layers.Layer):
    def __init__(self, units, **kwargs):
        # 先完成父层初始化，这是关键步骤
        super(DoubleDense, self).__init__(**kwargs)
        # 现在创建子层，name参数会正常生效
        self.dense1 = keras.layers.Dense(units, name="first_dense")
        self.dense2 = keras.layers.Dense(units, name="second_dense")
        
    def build(self, input_shape):
        # 修正build的输入形状：子层需要完整的形状元组，而非单一数值
        self.dense1.build(input_shape)
        # dense1的输出形状是(batch_size, units)，对应dense2的输入形状
        self.dense2.build((input_shape[0], self.dense1.units))
        
    def call(self, input):
        hidden = self.dense1(input)
        return self.dense2(hidden)

# 测试代码
dd = DoubleDense(3)
trivial_input = tf.ones((1,10))
output = dd(trivial_input)
print([weight.name for weight in dd.weights])

运行后输出的权重名称会变成：

['double_dense/first_dense/kernel:0', 'double_dense/first_dense/bias:0', 'double_dense/second_dense/kernel:0', 'double_dense/second_dense/bias:0']

修正方案二：显式注册子层（可选） #

如果你需要更明确地管理子层，可以使用self.add_layer()方法将子层注册到父层中，效果和方案一一致：

class DoubleDense(keras.layers.Layer):
    def __init__(self, units, **kwargs):
        super(DoubleDense, self).__init__(**kwargs)
        # 用add_layer显式注册子层
        self.dense1 = self.add_layer(keras.layers.Dense(units, name="first_dense"))
        self.dense2 = self.add_layer(keras.layers.Dense(units, name="second_dense"))
        
    def build(self, input_shape):
        self.dense1.build(input_shape)
        self.dense2.build((input_shape[0], self.dense1.units))
        
    def call(self, input):
        hidden = self.dense1(input)
        return self.dense2(hidden)

为什么原始代码不生效？ #

在你的原始代码中，子层是在super().__init__()之前创建的。此时父层还未完成初始化，Keras无法为子层设置父命名空间，导致子层的name参数被忽略，所有权重直接使用父层名称，最终出现重复的权重名。

只要确保父层先完成初始化，Keras就会自动处理子层的命名空间，让子层的name参数正常生效，权重名称自然会体现子层的归属。

内容的提问来源于stack exchange，提问作者Isaac Breen