TensorFlow 2/Keras函数式API训练时如何获取特定层权重？

我太懂你踩的这个坑了——你之前直接对Dense层输出的Tensor对象调用get_weights()，肯定会报错！因为get_weights()是Keras层（Layer类实例）的专属方法，而你写的A_DENSE = Dense(...) (INPUT)得到的是该层输出的张量，根本不是层本身的实例，这就是问题的核心所在。

可行解决方案：结合自定义Callback与层实例获取权重 #

针对你的需求，这里给两种实用思路，你可以根据场景选：

方法1：保留层实例，将权重直接传入自定义损失

如果你需要在损失计算里实时用到层权重，可以先把层的实例单独存下来，再把权重作为参数传给损失函数：

import tensorflow as tf
from tensorflow.keras.layers import Input, Dense
from tensorflow.keras.models import Model

# 先定义层实例，再用它构建模型
INPUT = Input(shape=(10,))
# 重点：把层实例单独保存，不要直接链式调用
a_dense_layer = Dense(1, use_bias=True, name="A_DENSE")
a_dense_output = a_dense_layer(INPUT)

# 自定义损失函数，接收层权重作为额外参数
def custom_loss(y_true, y_pred, layer_weights):
    # 拆分权重矩阵和偏置
    weight_matrix, bias = layer_weights
    # 这里可以把权重加入损失计算逻辑，比如加L2正则
    loss = tf.reduce_mean(tf.square(y_true - y_pred)) + tf.norm(weight_matrix)
    return loss

# 编译模型时，通过lambda把层权重传入损失
model = Model(inputs=INPUT, outputs=a_dense_output)
model.compile(
    optimizer='adam', 
    loss=lambda y_true, y_pred: custom_loss(y_true, y_pred, a_dense_layer.get_weights())
)

方法2：用自定义Callback监控/保存权重变化

如果你的需求是追踪训练过程中层权重的变化，或者不想修改损失函数的参数，可以用Callback在训练的关键节点获取权重：

class LayerWeightTracker(tf.keras.callbacks.Callback):
    def __init__(self, target_layer):
        super().__init__()
        self.target_layer = target_layer  # 传入目标层实例
        self.weight_history = []

    def on_epoch_end(self, epoch, logs=None):
        # 在每个epoch结束时获取当前层的权重
        current_weights = self.target_layer.get_weights()
        self.weight_history.append(current_weights)
        print(f"Epoch {epoch+1}: 已记录A_DENSE层的权重")

# 同样要先保留层实例
INPUT = Input(shape=(10,))
a_dense_layer = Dense(1, use_bias=True, name="A_DENSE")
a_dense_output = a_dense_layer(INPUT)
model = Model(inputs=INPUT, outputs=a_dense_output)

# 编译并训练，加入自定义Callback
model.compile(optimizer='adam', loss='mse')
model.fit(
    x_train, y_train, 
    epochs=10, 
    callbacks=[LayerWeightTracker(target_layer=a_dense_layer)]
)

# 训练结束后可以查看权重变化历史
print(LayerWeightTracker.weight_history)

核心要点提醒 #

• 务必区分层实例（比如a_dense_layer）和层输出张量（比如a_dense_output），只有层实例才能调用get_weights()。
• 若要在损失里实时用权重，选方法1；若要监控权重变化，方法2更灵活。

内容的提问来源于stack exchange，提问作者Applicable Math