tf.contrib.losses.cosine_distance

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

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.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/losses/cosine_distance