espnet.nets.pytorch_backend.transducer.blocks.build_transformer_block

source

espnet.nets.pytorch_backend.transducer.blocks.build_transformer_block(net_part: str, block: Dict[str, Any], pw_layer_type: str, pw_activation_type: str) → EncoderLayer | TransformerDecoderLayer

Build function for transformer block.

• Parameters:
  • net_part – Network part, either ‘encoder’ or ‘decoder’.
  • block – Transformer block parameters.
  • pw_layer_type – Positionwise layer type.
  • pw_activation_type – Positionwise activation type.
• Returns: Function to create transformer (encoder or decoder) block.