mean_reverse/mean_reverse.rs
1use onnx_helpers::prelude::*;
2use onnx_pb::{save_model, tensor_proto::DataType};
3
4fn main() {
5 let mut graph = builder::Graph::new("reverse");
6 let x = graph.input("X").typed(DataType::Float).dim(1).dim(6).node();
7 let two = graph.constant("two", 2.0f32);
8 let out = -(&x - x.mean(1, true)) * two + x;
9 let graph = graph.outputs_typed(out, DataType::Float);
10 let model = graph.model().build();
11 save_model("mean-reverse.onnx", &model).unwrap();
12}