阅读(9.1k) 书签 (0)

TensorFlow函数:tf.losses.add_loss

2018-08-18 15:24 更新

tf.losses.add_loss函数

tf.losses.add_loss(
    loss,
    loss_collection=tf.GraphKeys.LOSSES
)

定义在:tensorflow/python/ops/losses/util.py.

将外部定义的loss添加到losses的集合中.

参数:

  • loss:一个表示loss的Tensor.
  • loss_collection:一个可选的集合以添加loss.