快捷方式

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源