import onnx
from onnx import helper
from onnx import TensorProto

# Create a simple model
input_tensor = helper.make_tensor_value_info("input", TensorProto.FLOAT, [2, 3])
output_tensor = helper.make_tensor_value_info("output", TensorProto.FLOAT, [2, 3])
node = helper.make_node(
    "Identity",  # A simple identity operation
    inputs=["input"],
    outputs=["output"],
)

graph = helper.make_graph(
    nodes=[node],
    name="SimpleGraph",
    inputs=[input_tensor],
    outputs=[output_tensor],
)

model = helper.make_model(graph, producer_name="example-producer")

# Save the model to a file
model_path = "model.onnx"
onnx.save(model, model_path)