Method gather

utils/my_data_parallel.py:201–202

(self, outputs, output_device)

        return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])

    def gather(self, outputs, output_device):
        return gather(outputs, output_device, dim=self.dim)