def METHOD_NAME(self, states: Optional[torch.Tensor], idx: int) -> None:
    """Get specified ID state from decoder hidden states.
    Args:
        states: Decoder hidden states. None
        idx: State ID to extract.
    Returns:
        : Decoder hidden state for given ID. None
    """
    return None