onnx-helpers

ONNX graph construction helpers.

Usage

Toml [dependencies] onnx-helpers = "^1.0.0"

Example

```Rust use onnxhelpers::prelude::*; use onnxpb::tensor_proto::DataType;

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

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

let two = graph.constant(2.0f32);
let mean = x.mean(1, true);
let graph = graph.outputs(-(&x - mean) * two + x);
let model = graph.model().build();

save_model("mean-reverse.onnx", &model).unwrap();

} ```

Credits

Based on onnx-rs.

License

MIT license same as ONNX.