Method generate

dali/test/python/test_external_source_impl.py:131–138

(dim)

        random_seed(12345 * self.i + 4321)

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

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