Method __init__

tensorrt_llm/functional.py:1481–1485

(self, condition: Tensor)

    '''

    def __init__(self, condition: Tensor):
        self.layer = default_trtnet().add_if_conditional()
        if condition.ndim() > 0:
            condition = view(condition, [])
        self.layer.set_condition(condition.trt_tensor)

    def add_input(self, input: Tensor) -> Tensor:
        in_node = self.layer.add_input(input.trt_tensor)