一,tf.losses.add_loss 将外部定义的损失添加到损失集合中。
tf.losses.add_loss( loss, loss_collection=tf.GraphKeys.LOSSES)
参数:
loss:损耗张量。
loss_collection:将损失添加到其中的可选集合。
二,tf.losses.get_total_loss函数
tf.losses.get_total_loss( add_regularization_losses=True, name='total_loss')
参数:
add_regularization_losses:一个布尔值,指示是否在总和中使用正则化损失。
name:返回张量的名称。
三,例子
loss=tf.nn