我正在 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
检查docs https://keras.io/models/model/#get_layer您可以使用以下方法检索图层model.get_layer()
方法。然后,您可以传递所需的索引或传递图层的名称。
获得图层后,您可以使用以下命令轻松获得其输出layer.output
属性,如所解释的here https://keras.io/layers/about-keras-layers/在文档上。
将两者结合起来,您可以获得所需层的输出。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)