def METHOD_NAME(scores, index_to_token_maps):
    """Gets the argmax sequence from a set of scores.
    Args:
        scores (`list`): Sequences of output scores.
        index_to_token_maps (`list`): For each output token, maps
            the index in the probability distribution to a string.
    Returns:
        `list`: Representing the argmax sequence.
    """
    seq = []
    for score, tok_map in zip(scores, index_to_token_maps):
        # score_numpy_list = score.cpu().detach().numpy()
        score_numpy_list = score.cpu().numpy()
        assert score.shape[0] == len(tok_map) == len(list(score_numpy_list))
        seq.append(tok_map[np.argmax(score_numpy_list)])
    return seq