TensorFlow 2.0 Beta is available Learn more

tf.contrib.losses.compute_weighted_loss

Computes the weighted loss. (deprecated)

tf.contrib.losses.compute_weighted_loss(
    losses,
    weights=1.0,
    scope=None
)
View source on GitHub

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.