在 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 所述。

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