Function split_dataloader 

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

Available on crate features std and dataset only.

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