Method generate

dali/test/python/test_external_source_impl.py:673–678

(dim)

        random_seed(12345 * self.i + 4321)

        def generate(dim):
            shape = random_int(1, 10, [dim]).tolist()
            if self.as_tensor:
                return random_array([self.batch_size] + shape)
            else:
                return [random_array(shape) for _ in range(self.batch_size)]

        if self.i < self.n:
            self.i += 1