There was an error while loading. Please reload this page.
2 parents fc49f43 + cf0717b commit 3d1452cCopy full SHA for 3d1452c
tensorflow/python/ops/losses/losses_impl.py
@@ -652,7 +652,7 @@ def softmax_cross_entropy(
652
653
Args:
654
onehot_labels: `[batch_size, num_classes]` target one-hot-encoded labels.
655
- logits: [batch_size, num_classes] logits outputs of the network .
+ logits: `[batch_size, num_classes]` logits outputs of the network .
656
weights: Optional `Tensor` whose rank is either 0, or rank 1 and is
657
broadcastable to the loss which is a `Tensor` of shape `[batch_size]`.
658
label_smoothing: If greater than 0 then smooth the labels.
0 commit comments