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