src.parallel.scatter_gatherΒΆ
Functions
Gathers tensors from different GPUs on a specified device (-1 means the CPU). |
|
Slices tensors into approximately equal chunks and distributes them across given GPUs. |
|
Scatter with support for kwargs dictionary |
|
Scatter for customized sparse matrix |