MLXBackend.max_score_over_rotations#
- MLXBackend.max_score_over_rotations(scores, max_scores, rotations, rotation_index)#
Update elements in
max_scores
androtations
where scores is larger than max_scores with score and rotation_index, respectivelty.Warning
max_scores
androtations
are modified in-place.- Parameters:
- scoresBackendArray
The score space to compare against max_scores.
- max_scoresBackendArray
Maximum score observed for each element in an array.
- rotationsBackendArray
Rotation used to achieve a given max_score.
- rotation_indexint
The index representing the current rotation.
- Returns:
- Tuple[BackendArray, BackendArray]
Updated
max_scores
androtations
.