2个不稳定版本

0.2.0 2022年3月25日
0.1.0 2021年12月13日

#47 in #expect


用于 megenginelite-rs

Apache-2.0

7KB
158 代码行

megenginelite-rs

Crates.io libs.rs Documentation

Rust中的安全megenginelite包装器

⚠️ 项目仍处于早期开发阶段,预期会有错误、安全问题以及不工作的情况 ⚠️

安装

[dependencies]
megenginelite-rs = "1.8.2"

如何使用

// The dynamic library version needs to be greater than or equal to the compiled version.
// It is needless if the feature `auto-load` is enable (default enable).
unsafe {
    load("dynamic_library_path")?;
}

// set some options, and load model
let mut network = Network::builder()
        .dev_id(0)
        .stream_id(0)
        // ...
        .build("model_path")?;

// get an input of the model by name
let mut input = network.io_tensor("input_name").unwrap();
let data = Tensor::host()?;
input.copy_from(&data);

// exec, and wait
network.exec_wait()?;
// exec, async
network.exec().await?;

// get an output of the model by name
let output = network.io_tensor("output_name").unwrap();
println!("{:?}", output.as_slice::<f32>());

更多信息请查看 megenginelite

依赖项

~1.5MB
~35K SLoC