espnet2.asr.encoder.beats_encoder.TransformerEncoder

source

class espnet2.asr.encoder.beats_encoder.TransformerEncoder(config)

Bases: Module

Transformer encoder.

Initializes internal Module state, shared by both nn.Module and ScriptModule.

extract_features(x, padding_mask=None, tgt_layer=None)

Extract features from the input sequence.

forward(x, padding_mask=None, layer=None)

Forward pass.