gluonts.mx.block.mlp module#

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

Bases: 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[NDArray, Symbol]) Union[NDArray, 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