onnx.backend¶
后端¶
- class onnx.backend.base.Backend[source]¶
后端是接收具有输入的 ONNX 模型、执行计算，然后返回输出的实体。
对于一次性执行，用户可以使用 run_node 和 run_model 快速获取结果。
对于重复执行，用户应该使用 prepare，其中后端会完成执行模型所需的所有准备工作（例如，加载初始化器），并返回 BackendRep 处理程序。
- classmethod run_node(node: NodeProto, inputs: Any, device: str = 'CPU', outputs_info: Sequence[tuple[dtype, tuple[int, ...]]] | None = None, **kwargs: dict[str, Any]) tuple[Any, ...] | None[source]¶
简单运行一个运算符并返回结果。
- 参数:
node – 节点协议。
inputs – 节点的输入。
device – 要运行的设备。
outputs_info – 元组列表，包含每个输出的元素类型和形状。元组的第一个元素是 dtype，第二个元素是形状。更多用例可在 https://github.com/onnx/onnx/blob/main/onnx/backend/test/runner/__init__.py 中找到。
kwargs – 其他关键字参数。