gluonts.mx.representation.binning_helpers module¶
-
class
gluonts.mx.representation.binning_helpers.
Digitize
[source]¶ Bases:
mxnet.operator.CustomOp
-
backward
(req, out_grad, in_data, out_data, in_grad, aux)[source]¶ Backward interface. Can override when creating new operators.
- Parameters
req (list of str) – how to assign to in_grad. can be ‘null’, ‘write’, or ‘add’. You can optionally use self.assign(dst, req, src) to handle this.
in_data, out_data, in_grad, aux (out_grad,) – input and output for backward. See document for corresponding arguments of Operator::Backward
-
forward
(is_train, req, in_data, out_data, aux)[source]¶ Forward interface. Can override when creating new operators.
- Parameters
is_train (bool) – whether this is for training
req (list of str) – how to assign to out_data. can be ‘null’, ‘write’, or ‘add’. You can optionally use self.assign(dst, req, src) to handle this.
out_data, aux (in_data,) – input, output, and auxiliary states for forward. See document for corresponding arguments of Operator::Forward
-
-
class
gluonts.mx.representation.binning_helpers.
DigitizeProp
[source]¶ Bases:
mxnet.operator.CustomOpProp
-
create_operator
(ctx, in_shapes, in_dtypes)[source]¶ Create an operator that carries out the real computation given the context, input shapes, and input data types.
-
infer_shape
(in_shapes)[source]¶ infer_shape interface. Can override when creating new operators.
- Parameters
in_shape (list) – List of argument shapes in the same order as declared in list_arguments.
- Returns
in_shape (list) – List of argument shapes. Can be modified from in_shape.
out_shape (list) – List of output shapes calculated from in_shape, in the same order as declared in list_outputs.
aux_shape (Optional, list) – List of aux shapes calculated from in_shape, in the same order as declared in list_auxiliary_states.
-