Skip to content

Conversation

@sushreebarsa
Copy link
Contributor

@sushreebarsa sushreebarsa commented Jan 4, 2024

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!

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!
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants