def METHOD_NAME(
    self, y: torch.Tensor, state: Any, x: torch.Tensor
) -> Tuple[torch.Tensor, Any]:
    """Score new token.
    Args:
        y (torch.Tensor): 1D torch.int64 prefix tokens.
        state: Scorer state for prefix tokens
        x (torch.Tensor): encoder feature that generates ys.
    Returns:
        tuple[torch.Tensor, Any]: Tuple of
            torch.float32 scores for next token (vocab_size)
            and next state for ys
    """
    y = y.unsqueeze(0)
    h, _, cache = self.encoder.forward_one_step(
        self.embed(y), self._target_mask(y), cache=state
    )
    h = self.decoder(h[:, -1])
    logp = h.log_softmax(dim=-1).squeeze(0)
    return logp, cache