gluonts.mx.block.decoder 模块

class gluonts.mx.block.decoder.ForkingMLPDecoder(dec_len: int, final_dim: int, hidden_dimension_sequence: List[int] = [], **kwargs)[source]#

基类： gluonts.mx.block.decoder.Seq2SeqDecoder

用于序列到序列模型的多层感知机解码器。

详情请参阅 [WTN+17]。

参数

dec_len – 解码器长度（通常是预测的时间步数）。
final_dim – 每个时间步输出的维度（预测分位数的数量）。
hidden_dimension_sequence – 每个 MLP 层的隐藏单元数量。

hybrid_forward(F, static_input: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], dynamic_input: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]) → Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol][source]#

ForkingMLPDecoder 的前向传播调用。

参数

F – 在 MXNet 中可以引用 Symbol API 或 NDArray API 的模块。
static_input – 在此解码器中未使用。
dynamic_input – dynamic_features，形状为 (batch_size, sequence_length, num_features) 或 (N, T, C)，其中 sequence_length 等于编码器长度，num_features 对于例如 MQ-CNN 等于 channels_seq[-1] + 1 + decoder_length * num_feat_dynamic。

返回值

mlp 输出，形状为 (batch_size, sequence_length, decoder_length, decoder_mlp_dim_seq[0])。

返回类型

Tensor

class gluonts.mx.block.decoder.OneShotDecoder(decoder_length: int, layer_sizes: List[int], static_outputs_per_time_step: int)[source]#

基类： gluonts.mx.block.decoder.Seq2SeqDecoder

OneShotDecoder。

参数

decoder_length – 解码器长度（时间步数）
layer_sizes – 隐藏层的维度
static_outputs_per_time_step – 每个时间步的输出数量

hybrid_forward(F, static_input: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], dynamic_input: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]) → Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol][source]#

OneShotDecoder 的前向传播调用。

参数

F – 在 MXNet 中可以引用 Symbol API 或 NDArray API 的模块。
static_input – 静态特征，形状为 (batch_size, channels_seq[-1] + 1) 或 (N, C)
dynamic_input – dynamic_features，形状为 (batch_size, sequence_length, channels_seq[-1] + 1 + decoder_length * num_feat_dynamic) 或 (N, T, C)

返回值

mlp 输出，形状为 (batch_size, decoder_length, 最后一层大小)

返回类型

Tensor

class gluonts.mx.block.decoder.Seq2SeqDecoder(**kwargs)[source]#

基类： mxnet.gluon.block.HybridBlock

序列到序列模型中解码器块的抽象类。

hybrid_forward(F, static_input: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], dynamic_input: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]) → Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol][source]#

hybrid_forward 的抽象函数定义。

参数

static_input – 静态特征，形状为 (batch_size, channels_seq[-1] + 1) 或 (N, C)
dynamic_input – dynamic_features，形状为 (batch_size, sequence_length, channels_seq[-1] + 1 + decoder_length * num_feat_dynamic) 或 (N, T, C)

gluonts.mx.block.decoder 模块#