gluonts.mx.block.mlp 模块

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

基类：mxnet.gluon.block.HybridBlock

定义一个 MLP 块。

参数

• layer_sizes – 每层的隐藏单元数。

• flatten – 切换是否展平输出张量。

• activation – MLP 的激活函数，默认为 relu。

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

参数

• F – 可引用 MXNet 中的 Symbol API 或 NDArray API 的模块。

• x – 输入张量

返回

给定输入张量的 MLP 输出。

返回类型

张量