tf-keras
tf-keras copied to clipboard
Updating a typo in TF doc
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!