dgs.utils.state.collate_devices

dgs.utils.state.collate_devices(batch: list[torch.device], *_args, **_kwargs) → torch.device[source]

Collate a batch of devices into a single device.