src.parallel.scatter_gather

Functions

gather	Gathers tensors from different GPUs on a specified device (-1 means the CPU).
scatter	Slices tensors into approximately equal chunks and distributes them across given GPUs.
scatter_kwargs	Scatter with support for kwargs dictionary
scatter_sparse_matrix	Scatter for customized sparse matrix