python - 在keras自定义损失中使用层输出

Question

我正在 Keras 中开发自定义损失函数，我需要第一层输出。

如何找回它？

def custom_loss(y_true, y_pred):
    cross = K.mean(K.binary_crossentropy(y_true, y_pred), axis = 1)
    layer_output = model.get_layer_output(1) # this is what i'd like to use
    return cross  + perturb

Answer 1

检查docs您可以使用 model.get_layer() 方法检索图层。然后您可以传递所需的索引或传递图层的名称。

获取图层后，您可以使用layer.output属性轻松获取其输出，如所述 here在文档上。

将两者结合起来，您可以获得所需图层的输出。