gluonts.mx.model.transformer.trans_encoder module#

class gluonts.mx.model.transformer.trans_encoder.TransformerEncoder(encoder_length: int, config: Dict, **kwargs)[source]#

Bases: HybridBlock

hybrid_forward(F, data: Union[NDArray, Symbol]) → Union[NDArray, Symbol][source]#: A transformer encoder block consists of a self-attention and a feed- forward layer with pre/post process blocks in between.