File size: 248 Bytes
eef0ae4 | 1 2 3 4 5 6 7 | 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}")
|