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