gluonts.mx.batchify module

gluonts.mx.batchify.as_in_context(batch: dict, ctx: mxnet.context.Context = None) → Dict[str, Any][source]

Move data into new context, should only be in main process.

gluonts.mx.batchify.batchify(data: List[dict], ctx: Optional[mxnet.context.Context] = None, dtype: Optional[gluonts.core.component.DType] = <class 'numpy.float32'>, variable_length: bool = False) → Dict[str, Any][source]
gluonts.mx.batchify.stack(data, ctx: Optional[mxnet.context.Context] = None, dtype: Optional[gluonts.core.component.DType] = <class 'numpy.float32'>, variable_length: bool = False)[source]