TorchDynamo 的 ONNX 後端

有關 torch.compiler 的快速概述，請參閱torch.compiler。

警告

用於 torch.compile 的 ONNX 後端是一項快速發展的 Beta 技術。

torch.onnx.is_onnxrt_backend_supported()

如果已安裝 ONNX Runtime 相依性，且可用於支援 TorchDynamo 後端整合，則傳回 True；否則傳回 False。

範例

# xdoctest: +REQUIRES(env:TORCH_DOCTEST_ONNX)
>>> import torch
>>> if torch.onnx.is_onnxrt_backend_supported():
...     @torch.compile(backend="onnxrt")
...     def f(x):
...             return x * x
...     print(f(torch.randn(10)))
... else:
...     print("pip install onnx onnxscript onnxruntime")
...

傳回類型

bool