gluonts.mx.model.transformer.trans_encoder 模块#

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

基类: 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][source]#: 一个 Transformer 编码器块由一个自注意力层和一个前馈层组成，中间穿插了预处理/后处理块。