Function sample_gen

dali/test/python/test_dali_variable_batch_size.py:1467–1471

()

        return piepline(batch_size=max_batch_size, num_threads=4, device_id=0)

    def sample_gen():
        j = 42
        while True:
            yield np.full((13, 7), j)
            j += 1

    sample = sample_gen()
    batches = [