onnx推理之bind_ortvalue_output 的作用和功能

核心摘要

bind_ortvalue_output 是 ONNX Runtime 提供的一个 API 函数。它的主要作用是将一个 Ort::Value 对象绑定到模型推理的某个输出节点上。这样做的好处是,ONNX Runtime 会将模型的推理结果直接输出到您提供的这个 Ort::Value 中,而不是在内部创建一个新的 Ort::Value 再返回给你。

这通常用于避免不必要的数据拷贝,提升推理性能,尤其是在高性能或低延迟的应用场景中。


详细功能解释

要理解它的功能,我们最好对比一下使用和不使用它的区别。

1. 常规流程(不使用 bind_ortvalue_output

在常规的模型推理中,流程是这样的:

  1. 准备输入:创建 Ort::Value 对象,并填充好输入数据。

  2. 准备输出:通常只指定输出节点的名称,不提供具体的输出容器

  3. 运行推理:调用 Run 方法。

  4. 获取输出Run 方法会在内部为每一个输出节点新创建一个 Ort::Value 对象,并将计算结果填充进去,最后将这些对象返回给你。

cpp

// 
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值