o i @sddlZGdddeZdS)Nc@s4eZdZdZd ddZddZd dd Zd d ZdS)WindowStreamingE2EzWindowStreamingE2E constructor. :param E2E e2e: E2E ASR object :param recog_args: arguments for "recognize" method of E2E NcCsZ||_||_|j|_||_|jd|_d|_g|_g|_ d|_ |jj dks+JddS)NrgzGWindowStreamingE2E works only with combined CTC and attention decoders.) _e2e _recog_args char_list _char_list_rnnlmeval_offset!_previous_encoder_recurrent_state_encoder_states_ctc_posteriors_last_recognition ctc_weight)selfe2e recog_argsrnnlmr`/home/ubuntu/.local/lib/python3.10/site-packages/espnet/nets/pytorch_backend/streaming/window.py__init__ s zWindowStreamingE2E.__init__cCsb|j|\}}|j|d||j\}}|_|j|d|j|jj |ddS)z=Call this method each time a new batch of input is available.rN) rsubsample_framesenc unsqueezer r appendsqueezer ctc log_softmax)rxhilen_rrr accept_inputs   zWindowStreamingE2E.accept_inputFcsD|rtjjddtjjddfSfdd}|j|jfS)NrdimcsHj}d}g}|D]}||kr||q ||d7}q tj|ddS)Nrr")r rsizetorchcat)window_tensors last_offsetoffset_traversedselected_windowsesrrrselect_unprocessed_windows4s zPWindowStreamingE2E._input_window_for_decoder..select_unprocessed_windows)r&r'r r )ruse_allr.rr-r_input_window_for_decoder-s  z,WindowStreamingE2E._input_window_for_decodercCs,|jdd\}}|jj|||j|j|jS)aaRun the attention decoder offline. Works even if the previous layers (encoder and CTC decoder) were being run in the online mode. This method should be run after all the audio has been consumed. This is used mostly to compare the results between offline and online implementation of the previous layers. T)r/)r0rdecrecognize_beamrrr)rrlpzrrrdecode_with_attention_offlineDs z0WindowStreamingE2E.decode_with_attention_offline)N)F)__name__ __module__ __qualname____doc__rr!r0r4rrrrrs    r)r&objectrrrrrs