espnet.nets.pytorch_backend.transducer.blocks.prepare_input_layer

source

espnet.nets.pytorch_backend.transducer.blocks.prepare_input_layer(input_layer_type: str, feats_dim: int, blocks: List[Dict[str, Any]], dropout_rate: float, pos_enc_dropout_rate: float) → Dict[str, Any]

Prepare input layer arguments.

• Parameters:
  • input_layer_type – Input layer type.
  • feats_dim – Dimension of input features.
  • blocks – Blocks parameters for network part.
  • dropout_rate – Dropout rate for input layer.
  • pos_enc_dropout_rate – Dropout rate for input layer pos. enc.
• Returns: Input block parameters.
• Return type: input_block