onnx-helpers

ONNX图构建助手。

使用方法

[dependencies]
onnx-helpers = { git = "https://github.com/crackcomm/onnx-helpers-rs.git" }

示例

use onnx_helpers::prelude::*;
use onnx_pb::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("two", 2.0f32);
    let graph = graph.outputs(-(&x - x.mean(1, true)) * two + x);
    let model = graph.model().build();
    save_model("mean-reverse.onnx", &model).unwrap();
}

鸣谢

基于onnx-rs。

许可

与ONNX相同的MIT许可。

lib.rs:

ONNX模型构建助手。