ONNX бэкенд для TorchDynamo

Для быстрого обзора torch.compiler, см. torch.compiler.

torch.onnx.is_onnxrt_backend_supported()

Возвращает True , если зависимости ONNX Runtime установлены и пригодны для интеграции бэкенда TorchDynamo; 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-preview onnxruntime")
...

Тип возвращаемого значения

bool