use onnx_helpers::prelude::*;
use onnx_pb::{save_model, tensor_proto::DataType};

fn main() {
    let mut graph = builder::Graph::new("add");

    let x = graph
        .input("X")
        .typed(DataType::Float)
        .dim(1)
        .dim(10)
        .node();
    let y = graph
        .input("Y")
        .typed(DataType::Float)
        .dim(1)
        .dim(10)
        .node();

    let z = (x + y).with_name("Z");

    let model = graph.outputs(z).model().build();

    save_model("add.onnx", &model).unwrap();
}