Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.losses.compute_weighted_loss

View source on GitHub

Computes the weighted loss. (deprecated)

tf.contrib.losses.compute_weighted_loss(
    losses,
    weights=1.0,
    scope=None
)

Args:

  • losses: A tensor of size [batch_size, d1, ... dN].
  • weights: A tensor of size [1] or [batch_size, d1, ... dK] where K < N.
  • scope: the scope for the operations performed in computing the loss.

Returns:

A scalar Tensor that returns the weighted loss.

Raises:

  • ValueError: If weights is None or the shape is not compatible with losses, or if the number of dimensions (rank) of either losses or weights is missing.