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

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