espnet.nets.pytorch_backend.transducer.utils.check_state

source

espnet.nets.pytorch_backend.transducer.utils.check_state(state: List[Tensor | None], max_len: int, pad_id: int) → List[Tensor | None]

Check decoder hidden states and left pad or trim if necessary.

• Parameters:
  • state – Decoder hidden states. [N x (?, D_dec)]
  • max_len – maximum sequence length.
  • pad_id – Padding symbol ID.
• Returns: Decoder hidden states. [N x (1, max_len, D_dec)]
• Return type: final