Function pipeline

dali/test/python/conditionals/test_nests.py:64–72

(unpacking_select)

    @pipeline_def(enable_conditionals=True, num_threads=4, batch_size=8, device_id=0)
    def pipeline(unpacking_select):
        image = types.Constant(np.full((200, 300, 3), 42, dtype=np.uint8), device="cpu")
        label = types.Constant(np.array(1), device="cpu")
        ops = [rotate, color]
        op_idx = fn.random.uniform(values=list(range(len(ops))))
        image, label = select(
            ops, op_idx, image=image, label=label, unpacking_select=unpacking_select
        )
        return image, label

    pipe_unpacking = pipeline(unpacking_select=True)
    pipe_unpacking.run()