tf.contrib.losses.cosine_distance
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 . |
© 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