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") ...
- 返回类型