ONNX 后端用于 TorchDynamo ¶

想快速了解 torch.compiler ，请参阅 torch.compiler。

警告

ONNX 后端为 torch.compile 是一种快速发展的测试版技术。

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")
...

返回类型:

布尔型