gluonts.mx.representation.dim_expansion 模块#

class gluonts.mx.representation.dim_expansion.DimExpansion(axis: int = - 1, *args, **kwargs)[源代码]#

基类：gluonts.mx.representation.representation.Representation

表示沿指定轴进行维度扩展操作的类。

参数

axis – 张量进行维度扩展的轴。（默认值：-1）

hybrid_forward(F, data: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], observed_indicator: Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], scale: Optional[Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]], rep_params: List[Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]], **kwargs) → Tuple[Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol], List[Union[mxnet.ndarray.ndarray.NDArray, mxnet.symbol.symbol.Symbol]]][源代码]#

将数据转换为所需的表示形式。

参数

  • F –

  • data – 目标数据。

  • observed_indicator – 目标观测指示器。

  • scale – 预计算的比例因子。

  • rep_params – 其他预计算的表示参数。

  • **kwargs – 其他块特定参数。

：参数：其他块特定参数。

返回值

包含转换后的数据、计算出的比例因子以及要传递给 post_transform 的附加参数的元组。

返回类型

Tuple[Tensor, Tensor, List[Tensor]]