tf-keras icon indicating copy to clipboard operation
tf-keras copied to clipboard

Updating a typo in TF doc

Open sushreebarsa opened this issue 2 years ago • 0 comments

Updated the line from if (y_pred_rank - y_true_rank != 1) or y_pred_shape[-1] == 1: to if (y_pred_rank - y_true_rank == 1) or y_pred_shape[-1] == 1: as the current behavior will squeeze the y_pred tensor even when it’s the same rank and shape as the y_pred tensor. Fixes #62718

Please have a look at this and do the needful. Thank you!

sushreebarsa avatar Jan 04 '24 06:01 sushreebarsa