espnet.nets.pytorch_backend.transducer.utils.check_batch_states

source

espnet.nets.pytorch_backend.transducer.utils.check_batch_states(states, max_len, pad_id)

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

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