def METHOD_NAME(self, y, ids, state, x):
    """Score new token.
    Args:
        y (torch.Tensor): 1D prefix token
        ids (torch.Tensor): torch.int64 next token to score
        state: decoder state for prefix tokens
        x (torch.Tensor): 2D encoder feature that generates ys
    Returns:
        tuple[torch.Tensor, Any]:
            Tuple of a score tensor for y that has a shape `(len(next_tokens),)`
            and next state for ys
    """
    batch_state = (
        (
            torch.stack([s[0] for s in state], dim=2),
            torch.stack([s[1] for s in state]),
            state[0][2],
            state[0][3],
        )
        if state[0] is not None
        else None
    )
    return self.impl(y, batch_state, ids)