Fixing arithmetic expression in comment.
PiperOrigin-RevId: 268178237
This commit is contained in:
parent
5800fe4812
commit
b266a849d8
@ -4406,7 +4406,7 @@ def sparse_categorical_crossentropy(target, output, from_logits=False, axis=-1):
|
|||||||
|
|
||||||
target = cast(target, 'int64')
|
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)
|
output_shape = array_ops.shape_v2(output)
|
||||||
target_rank = target.shape.ndims
|
target_rank = target.shape.ndims
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user