快捷方式

TorchScript 不支持的 PyTorch 结构

Torch 和张量不支持的属性

TorchScript 支持在 torchtorch.Tensor 上定义的大多数方法,但我们没有完全覆盖。以下是 Python 和 TorchScript 之间行为不同的特定已知运算符和运算符类别。如果您遇到其他不支持的操作,请在 GitHub 上提交问题。已弃用的运算符不在下面列出。

不支持的张量方法

不支持的张量属性

在 Torch 上未正确绑定的函数

如果在 TorchScript 中使用以下函数,它们将失败,因为它们未绑定到 torch,或者因为 Python 期望的模式与 TorchScript 不同。

Torch 和 Python 之间模式不同的运算符

以下类别的运算符具有不同的模式

从非张量输入构建张量的函数不支持 requires_grad 参数,除了 torch.tensor。这涵盖以下运算符

以下函数在 TorchScript 中需要 dtypelayoutdevice 作为参数,但在 Python 中这些参数是可选的。

PyTorch 不支持的模块和类

TorchScript 目前无法编译许多其他常用的 PyTorch 结构。下面列出了 TorchScript 不支持的模块,以及不支持的 PyTorch 类的不完全列表。对于不支持的模块,我们建议使用 torch.jit.trace().

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源