gluonts.model.transformer.trans_encoder module

class gluonts.model.transformer.trans_encoder.TransformerEncoder(encoder_length: int, config: Dict, **kwargs)

Bases: mxnet.gluon.block.HybridBlock

hybrid_forward(F, data: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]) → Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]

A transformer encoder block consists of a self-attention and a feed-forward layer with pre/post process blocks in between.