Function _num_elements

tensorflow/python/ops/losses/losses_impl.py:132–135

Computes the number of elements in `losses` tensor.

(losses)

def _num_elements(losses):
  """Computes the number of elements in `losses` tensor."""
  with ops.name_scope(None, "num_elements", values=[losses]) as scope:
    return math_ops.cast(array_ops.size(losses, name=scope), dtype=losses.dtype)


@tf_export(v1=["losses.compute_weighted_loss"])