Function select

dali/test/python/conditionals/test_nests.py:49–53

(ops, selected_op_idx, *op_args, unpacking_select=False, **op_kwargs)

        return a, b

    def select(ops, selected_op_idx, *op_args, unpacking_select=False, **op_kwargs):
        if unpacking_select:
            return _select_unpack(0, len(ops) - 1, ops, selected_op_idx, op_args, op_kwargs)
        else:
            return _select_fwd(0, len(ops) - 1, ops, selected_op_idx, op_args, op_kwargs)

    def rotate(image, label):
        image = fn.rotate(image, angle=42)