2个不稳定版本
0.2.0 | 2022年3月25日 |
---|---|
0.1.0 | 2021年12月13日 |
#47 in #expect
7KB
158 代码行
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。
依赖项
~1.5MB
~35K SLoC