| 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}") | |
| 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}") | |