gluonts.torch.batchify module

gluonts.torch.batchify.batchify(data: List[dict], device: Optional[Union[device, str, int]] = None) → Dict[str, Any]

gluonts.torch.batchify.stack(data, device: Optional[Union[device, str, int]] = None)