gluonts.mx.block.snmlp 模块

class gluonts.mx.block.snmlp.SNMLPBlock(in_units: int, hidden_units: int, out_units: int, num_hidden_layers: int = 2, activation: str = 'lipswish', jacobian_method: str = 'bf', num_power_iter: int = 1, coeff: float = 0.9, flatten: bool = False)

Bases: mxnet.gluon.block.HybridBlock

get_weights()

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

参数

x – 输入张量

返回值

SNMLPBlock 的输出

返回类型

张量

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

参数

x – 输入张量

返回值

SNMLPBlock 在 x 处的雅可比行列式。

返回类型

张量

gluonts.mx.block.snmlp.jacobian_sn_mlp_block_bf(layers: List[Tuple[mxnet.gluon.block.HybridBlock, Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]]]) → Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]

SNMLPBlock 雅可比行列式的暴力计算，其形状为 (Batch dim1, …, Output dim, Input dim)

参数

layers – 一个元组列表，每个元组 (layer, input) 与 SNMLPBlock 的组成层相关联，其中 layer 对应于相关的对象层及其输入张量。

返回值

在给定输入下计算的 SNMLPBlock 雅可比行列式

返回类型

张量