Skip to content

Commit 3d1452c

Browse files
authored
Merge pull request tensorflow#14930 from mrshu/mrshu-softmax_cross_entropy
softmax_cross_entropy: Improve docstring
2 parents fc49f43 + cf0717b commit 3d1452c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow/python/ops/losses/losses_impl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -652,7 +652,7 @@ def softmax_cross_entropy(
652652
653653
Args:
654654
onehot_labels: `[batch_size, num_classes]` target one-hot-encoded labels.
655-
logits: [batch_size, num_classes] logits outputs of the network .
655+
logits: `[batch_size, num_classes]` logits outputs of the network .
656656
weights: Optional `Tensor` whose rank is either 0, or rank 1 and is
657657
broadcastable to the loss which is a `Tensor` of shape `[batch_size]`.
658658
label_smoothing: If greater than 0 then smooth the labels.

0 commit comments

Comments
 (0)