- nb_correct += (
- (
- (new_quizzes == result).long()
- * (inverted_quizzes, inverted_result).long()
- )
- .min(dim=-1)
- .values
- )
+ nb_correct += (new_quizzes == result).long().min(dim=-1).values * (
+ inverted_quizzes == inverted_result
+ ).long().min(dim=-1).values