TensorFlow函数:tf.losses.add_loss

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.
以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

意见反馈
返回顶部