Struct Model

Source

pub struct Model { /* private fields */ }

Expand description

Model builder.

Implementations§

Source §

impl Model

Source

pub fn new<G: Into<GraphProto>>(graph: G) -> Self

Creates a new builder.

Source

pub fn domain<S: Into<String>>(self, domain: S) -> Self

Sets model doc_string.

Source

pub fn model_version(self, model_version: i64) -> Self

Sets model doc_string.

Source

pub fn producer_name<S: Into<String>>(self, producer_name: S) -> Self

Sets model doc_string.

Source

pub fn producer_version<S: Into<String>>(self, producer_version: S) -> Self

Sets model doc_string.

Source

pub fn doc_string<S: Into<String>>(self, doc_string: S) -> Self

Sets model doc_string.

Source

pub fn metadata<K: Into<String>, V: Into<String>>( self, key: K, value: V, ) -> Self

Inserts model metadata.

Source

pub fn opset_import(self, opset: OperatorSetIdProto) -> Self

Inserts operator set import.

Source

pub fn build(self) -> ModelProto

Builds the model.

Examples found in repository ?

examples/mean_reverse.rs (line 10)

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}

More examples

Hide additional examples

examples/stddev.rs (line 10)

4fn main() {
5    let mut graph = builder::Graph::new("stddev");
6    let x = graph.input("X").typed(DataType::Float).dim(1).dim(6).node();
7    let two = graph.constant("two", 2.0f32);
8    let std = (&x - x.mean(1, true)).abs().pow(two).mean(1, true).sqrt();
9    let graph = graph.outputs_typed(std.with_name("stddev"), DataType::Float);
10    let model = graph.model().build();
11    save_model("stddev.onnx", &model).unwrap();
12}

examples/add.rs (line 22)

4fn main() {
5    let mut graph = builder::Graph::new("add");
6
7    let x = graph
8        .input("X")
9        .typed(DataType::Float)
10        .dim(1)
11        .dim(10)
12        .node();
13    let y = graph
14        .input("Y")
15        .typed(DataType::Float)
16        .dim(1)
17        .dim(10)
18        .node();
19
20    let z = (x + y).with_name("Z");
21
22    let model = graph.outputs(z).model().build();
23
24    save_model("add.onnx", &model).unwrap();
25}

examples/concat.rs (line 11)

4fn main() {
5    let mut graph = builder::Graph::new("concat");
6    let x = graph.input("X").typed(DataType::Float).dim(1).dim(6).node();
7    let two = graph.constant("two", 2.0f32);
8    let mean_reverse = -(&x - x.mean(1, true)) * two + &x;
9    let concat = graph.concat(0, vec![x, mean_reverse]).with_name("concat");
10    let graph = graph.outputs_typed(concat, DataType::Float);
11    let model = graph.model().build();
12    save_model("concat.onnx", &model).unwrap();
13}

examples/ensemble.rs (line 12)

4fn main() {
5    let mut graph = builder::Graph::new("stddev");
6    let x = graph.input("X").typed(DataType::Float).dim(1).dim(6).node();
7    let std = stddev(&mut graph, &x);
8    let mrev = mean_reverse(&mut graph, &x);
9    let graph = graph
10        .outputs_typed(std.with_name("stddev"), DataType::Float)
11        .outputs_typed(mrev.with_name("mean_reverse"), DataType::Float);
12    let model = graph.model().build();
13    save_model("ensemble.onnx", &model).unwrap();
14}