在keras自定义损失中使用图层输出 [英] Use layer output in keras custom loss

问题描述

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

I am developing a custom loss function in Keras and I need the first layer output.

我如何找回它?

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

推荐答案

检查文档您可以使用model.get_layer()方法检索图层.然后，您可以传递所需的索引，也可以传递层的名称.

Checking the docs you can retrieve a layer by using the model.get_layer() method. You can then pass the desired index or well pass the name of the layer.

获取图层后，您可以使用layer.output属性轻松获取其输出，如

After getting a layer you can easily obtain its output by using the layer.output attribute, as explained here on the docs.

将两者结合起来就可以得到所需图层的输出.

Combining both you can obtain the output of your desired layer.

这篇关于在keras自定义损失中使用图层输出的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！