onnx.backend¶
后端¶
- class onnx.backend.base.Backend[source]¶
后端是一个实体，它将接收带有输入的ONNX模型，执行计算，然后返回输出。
对于一次性执行，用户可以使用run_node和run_model快速获取结果。
对于重复执行，用户应使用prepare，其中Backend会为重复执行模型完成所有准备工作（例如，加载初始化器），并返回一个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]¶
简单运行一个操作符并返回结果。
- Parameters:
node – 节点原型。
inputs – 节点的输入。
device – 运行的设备。
outputs_info – 一个包含每个输出元素类型和形状的元组列表。元组的第一个元素是数据类型，第二个元素是形状。更多用例可以在https://github.com/onnx/onnx/blob/main/onnx/backend/test/runner/__init__.py找到。
kwargs – 其他关键字参数。