Function unroll_input

monai/networks/trt_compiler.py:241–252

(input_names, input_example)

def unroll_input(input_names, input_example):
    # Simulate list/tuple unrolling during ONNX export
    unrolled_input = {}
    for name in input_names:
        val = input_example[name]
        if val is not None:
            if isinstance(val, list) or isinstance(val, tuple):
                for i in range(len(val)):
                    unrolled_input[f"{name}_{i}"] = make_tensor(val[i])
            else:
                unrolled_input[name] = make_tensor(val)
    return unrolled_input


def parse_groups(