mmengine.dist.cast_data_device¶ mmengine.dist.cast_data_device(data, device, out=None)[source]¶ Recursively convert Tensor in data to device. If data has already on the device, it will not be casted again. Parameters data (Tensor or list or dict) – Inputs to be casted. device (torch.device) – Destination device type. out (Tensor or list or dict, optional) – If out is specified, its value will be equal to data. Defaults to None. Returns data was casted to device. Return type Tensor or list or dict