src.parallel.scatter_gather.scatter

src.parallel.scatter_gather.scatter(inputs, target_gpus, dim=0)

Slices tensors into approximately equal chunks and distributes them across given GPUs. Duplicates references to objects that are not tensors.