espnet2.asr_transducer.beam_search_transducer.ExtendedHypothesis

source

class espnet2.asr_transducer.beam_search_transducer.ExtendedHypothesis(score: float, yseq: List[int], dec_state: Tuple[Tensor, Tensor | None] | None = None, lm_state: Dict[str, Any] | List[Any] | None = None, dec_out: Tensor | None = None, lm_score: Tensor | None = None)

Bases: Hypothesis

Extended hypothesis definition for NSC beam search and mAES.

:param : Hypothesis dataclass arguments. :param dec_out: Decoder output sequence. (B, D_dec) :param lm_score: Log-probabilities of the LM for given label. (vocab_size)

dec_out : Tensor = None

lm_score : Tensor = None