onnx.backend¶
後端¶
- class onnx.backend.base.Backend[原始碼]¶
後端是將 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[原始碼]¶
簡單執行一個運算符並傳回結果。
- 參數:
node – 節點原型。
inputs – 節點的輸入。
device – 要在其上執行的裝置。
outputs_info – 包含每個輸出的元素類型和形狀的元組清單。元組的第一個元素是 dtype，第二個元素是形狀。在 https://github.com/onnx/onnx/blob/main/onnx/backend/test/runner/__init__.py 中可以找到更多使用案例
kwargs – 其他關鍵字參數。