gluonts.mx.block.mlp module

class gluonts.mx.block.mlp.MLP(layer_sizes: List[int], flatten: bool, activation='relu')[source]

Bases: mxnet.gluon.block.HybridBlock

Defines an MLP block.

Parameters
  • layer_sizes – number of hidden units per layer.

  • flatten – toggle whether to flatten the output tensor.

  • activation – activation function of the MLP, default is relu.

hybrid_forward(F, x: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]) → Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol][source]
Parameters
  • F – A module that can either refer to the Symbol API or the NDArray API in MXNet.

  • x – Input tensor

Returns

Output of the MLP given the input tensor.

Return type

Tensor