JaxBackend.max_score_over_rotations#

JaxBackend.max_score_over_rotations(scores, max_scores, rotations, rotation_index)[source]#

Update elements in max_scores and rotations where scores is larger than max_scores with score and rotation_index, respectivelty.

Warning

max_scores and rotations 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 and rotations.