View source on GitHub |
Adds a externally defined loss to the collection of losses.
tf.losses.add_loss(
loss, loss_collection=tf.GraphKeys.LOSSES
)
Args | |
---|---|
loss
|
A loss Tensor .
|
loss_collection
|
Optional collection to add the loss to. |