Fixing arithmetic expression in comment.

PiperOrigin-RevId: 268178237
This commit is contained in:
A. Unique TensorFlower 2019-09-10 02:04:08 -07:00 committed by TensorFlower Gardener
parent 5800fe4812
commit b266a849d8

View File

@ -4406,7 +4406,7 @@ def sparse_categorical_crossentropy(target, output, from_logits=False, axis=-1):
target = cast(target, 'int64')
# Try to adjust the shape so that rank of labels = 1 - rank of logits.
# Try to adjust the shape so that rank of labels = rank of logits - 1.
output_shape = array_ops.shape_v2(output)
target_rank = target.shape.ndims