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]
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)