TensorFlow函數(shù)：tf.losses.add_loss

2018-08-18 15:24 更新

tf.losses.add_loss函數(shù)

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

定義在：tensorflow/python/ops/losses/util.py.

將外部定義的loss添加到losses的集合中.

參數(shù)：

  • loss：一個表示loss的Tensor.
  • loss_collection：一個可選的集合以添加loss.
以上內容是否對您有幫助：
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號