diff options
Diffstat (limited to 'tensorflow/python/ops/losses/losses.py')
-rw-r--r-- | tensorflow/python/ops/losses/losses.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/ops/losses/losses.py b/tensorflow/python/ops/losses/losses.py index a2f5d1d375..9da1def981 100644 --- a/tensorflow/python/ops/losses/losses.py +++ b/tensorflow/python/ops/losses/losses.py @@ -603,7 +603,7 @@ def softmax_cross_entropy( loss_collection: collection to which the loss will be added. Returns: - A scalar `Tensor` representing the loss value. + A scalar `Tensor` representing the mean loss value. Raises: ValueError: If the shape of `logits` doesn't match that of `onehot_labels` @@ -652,7 +652,7 @@ def sparse_softmax_cross_entropy(labels, logits, weights=1.0, scope=None, loss_collection: collection to which the loss will be added. Returns: - A scalar `Tensor` representing the loss value. + A scalar `Tensor` representing the mean loss value. Raises: ValueError: If the shapes of logits, labels, and weight are incompatible, or |