move_to_device

pytorch_forecasting.utils.move_to_device(x: Union[Dict[str, Union[Tensor, List[Tensor], Tuple[Tensor]]], Tensor, List[Tensor], Tuple[Tensor]], device: Union[str, DeviceObjType]) → Union[Dict[str, Union[Tensor, List[Tensor], Tuple[Tensor]]], Tensor, List[Tensor], Tuple[Tensor]]

Move object to device.

Parameters

• x (dictionary of list of tensors) – object (e.g. dictionary) of tensors to move to device

• device (Union[str, torch.DeviceObjType]) – device, e.g. “cpu”

Returns

x on targeted device