TensorFlow函数:tf.losses.add_loss

由 Carrie 创建, 最后一次修改 2018-08-18

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.
以上内容是否对您有帮助:
二维码
建议反馈
二维码