src.parallel.scatter_gather.gather¶

src.parallel.scatter_gather.gather(outputs, target_device, dim=0)[source]¶: Gathers tensors from different GPUs on a specified device (-1 means the CPU).