Python Dense.add_loss示例

示例#1

文件： test_topology.py 项目： zqy0/keras

def test_get_losses_for():
    a = Input(shape=(2,))
    dense_layer = Dense(1)
    dense_layer.add_loss(0, inputs=a)
    dense_layer.add_loss(1, inputs=None)

    assert dense_layer.get_losses_for(a) == [0]
    assert dense_layer.get_losses_for(None) == [1]

示例#2

def test_get_losses_for():
    a = Input(shape=(2, ))
    dense_layer = Dense(1)
    dense_layer.add_loss(lambda: 0, inputs=a)
    dense_layer.add_loss(lambda: 1, inputs=None)