TensorFlow函數(shù):tf.losses.get_total_loss

2018-08-23 11:30 更新

tf.losses.get_total_loss函數(shù)

tf.losses.get_total_loss(
    add_regularization_losses=True,
    name='total_loss'
)

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

返回一個張量,其值代表總loss.

特別的是,這增加了您使用tf.add_loss()在層構(gòu)造器上通過正則化參數(shù)添加的任何正則化損失所增加的任何loss,例如,tf.layers.如果您手動構(gòu)建一個loss_op,請務必使用此選項.否則關(guān)于tf.layers方法的正則化參數(shù)將無法正常工作.

參數(shù):

  • add_regularization_losses:一個布爾值,指示是否在總和中使用正則化損失.
  • name:返回張量的名稱.

返回:

返回一個Tensor,它的值代表總loss.

可能引發(fā)的異常:

  • ValueError:如果losses不可迭代.
以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號