head.py 文件源码

def _weighted_loss(loss, weight):
  """Returns cumulative weighted loss as 1d `Tensor`."""
  with ops.name_scope(None, "weighted_loss", (loss, weight)) as name:
    return math_ops.multiply(
        array_ops.reshape(loss, shape=(-1,)),
        array_ops.reshape(weight, shape=(-1,)),
        name=name)