Lib.rs

megenginelite-rs

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。