Method construct_even_inputs

test/distributed/algorithms/test_join.py:187–191

r"""Returns even inputs: each rank gets ``base`` inputs.

(self, base, device=None)

        return [torch.zeros(1, device=device) for _ in range(base + self.rank * offset)]

    def construct_even_inputs(self, base, device=None):
        r"""Returns even inputs: each rank gets ``base`` inputs."""
        if device is None:
            device = self.device
        return [torch.zeros(1, device=device) for _ in range(base)]

    @property
    def base_num_inputs(self):