Function split_dataloader 

pub fn split_dataloader<O>(
    dataloader: Arc<dyn DataLoader<O>>,
    devices: &[Device],
) -> Vec<Arc<dyn DataLoader<O>>>

Splits a dataloader into multiple partial dataloaders (one per device).