TensorFlow 2.0 Beta is available Learn more

tf.contrib.losses.cosine_distance

View source on GitHub

Adds a cosine-distance loss to the training procedure. (deprecated arguments) (deprecated)

tf.contrib.losses.cosine_distance(
    predictions,
    labels=None,
    axis=None,
    weights=1.0,
    scope=None,
    dim=None
)

Note that the function assumes that predictions and labels are already unit-normalized.

Args:

  • predictions: An arbitrary matrix.
  • labels: A Tensor whose shape matches 'predictions'
  • axis: The dimension along which the cosine distance is computed.
  • weights: Coefficients for the loss a scalar, a tensor of shape [batch_size] or a tensor whose shape matches predictions.
  • scope: The scope for the operations performed in computing the loss.
  • dim: The old (deprecated) name for axis.

Returns:

A scalar Tensor representing the loss value.

Raises:

  • ValueError: If predictions shape doesn't match labels shape, or weights is None.