gluonts.mx.block.sndense 模块

class gluonts.mx.block.sndense.SNDense(units: int, in_units: int, coeff: float = 0.9, activation: typing.Optional[str] = None, use_bias: bool = True, flatten: bool = True, weight_initializer: mxnet.initializer.Initializer = <mxnet.initializer.Orthogonal object>, bias_initializer='zeros', dtype='float32', num_power_iter: int = 1, ctx: typing.Optional[mxnet.context.Context] = None, **kwargs)

基类: mxnet.gluon.block.HybridBlock

应用于权重谱范数归一化的 Dense 层，如 [BJC19] 所示。

[BJC19].

hybrid_forward(F, x, _weight, _u, _bias=None)

参数

x – SNDense 层的输入张量

返回

SNDense 层的输出张量

返回类型

Tensor

property weight