import onnx model = onnx.load("onnx/frida-onnx/FRIDA.onnx") for inp in model.graph.input: shape = [dim.dim_value if dim.dim_value > 0 else str(dim.dim_param) for dim in inp.type.tensor_type.shape.dim] print(f"Input '{inp.name}': {shape}")