PytorchBackend.max_score_over_rotations

PytorchBackend.max_score_over_rotations(scores, max_scores, rotations, rotation_index)

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.