o ¡¿¯i ã@sNdZddlmZmZddlmZmZmZmZddl Z Gdd„de j j eƒZ dS)z2Abstract decoder definition for Transducer models.é)ÚABCÚabstractmethod)ÚAnyÚListÚOptionalÚTupleNc@sBeZdZdZedejdejfdd„ƒZedejdee de e eje ejfde eje e eje ejfffd d „ƒZ ed ee de eje e eje ejfffd d „ƒZedejddfdd„ƒZede de e eje ejffdd„ƒZe  dde e eje ejfde de e eje ejffdd„ƒZdS)Ú AbsDecoderzAbstract decoder module.ÚlabelsÚreturncCót‚)zµEncode source label sequences. Args: labels: Label ID sequences. (B, L) Returns: dec_out: Decoder output sequences. (B, T, D_dec) ©ÚNotImplementedError)Úselfr ©rú^/home/ubuntu/.local/lib/python3.10/site-packages/espnet2/asr_transducer/decoder/abs_decoder.pyÚforward s zAbsDecoder.forwardÚlabelÚlabel_sequenceÚ dec_statecCr )aáOne-step forward hypothesis. Args: label: Previous label. (1, 1) label_sequence: Current label sequence. dec_state: Previous decoder hidden states. ((N, 1, D_dec), (N, 1, D_dec) or None) or None Returns: dec_out: Decoder output sequence. (1, D_dec) or (1, D_emb) dec_state: Decoder hidden states. ((N, 1, D_dec), (N, 1, D_dec) or None) or None r )rrrrrrrÚscoreszAbsDecoder.scoreÚhypscCr )aOne-step forward hypotheses. Args: hyps: Hypotheses. Returns: dec_out: Decoder output sequences. (B, D_dec) or (B, D_emb) states: Decoder hidden states. ((N, B, D_dec), (N, B, D_dec) or None) or None r )rrrrrÚ batch_score0szAbsDecoder.batch_scoreÚdeviceNcCr )zNSet GPU device to use. Args: device: Device ID. r )rrrrrÚ set_deviceBszAbsDecoder.set_deviceÚ batch_sizecCr )zÕInitialize decoder states. Args: batch_size: Batch size. Returns: : Initial decoder hidden states. ((N, B, D_dec), (N, B, D_dec) or None) or None r )rrrrrÚ init_stateLszAbsDecoder.init_staterÚstatesÚidxcCr )ajGet specified ID state from batch of states, if provided. Args: states: Decoder hidden states. ((N, B, D_dec), (N, B, D_dec) or None) or None idx: State ID to extract. Returns: : Decoder hidden state for given ID. ((N, 1, D_dec), (N, 1, D_dec) or None) or None r )rrrrrrÚ select_state\szAbsDecoder.select_state)Nr)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚtorchÚTensorrrÚintrrrrrrÚtensorrrrrrrr sL þýü  ûþ  ý ÿ þýþýür) r"ÚabcrrÚtypingrrrrr#ÚnnÚModulerrrrrÚs