TorchScript 不支持的 PyTorch 结构¶
Torch 和张量不支持的属性¶
TorchScript 支持在 torch
和 torch.Tensor
上定义的大多数方法,但我们没有完全覆盖。以下是 Python 和 TorchScript 之间行为不同的特定已知运算符和运算符类别。如果您遇到其他不支持的操作,请在 GitHub 上提交问题。已弃用的运算符不在下面列出。
不支持的张量方法¶
不支持的张量属性¶
Torch 和 Python 之间模式不同的运算符¶
以下类别的运算符具有不同的模式
从非张量输入构建张量的函数不支持 requires_grad 参数,除了 torch.tensor。这涵盖以下运算符
以下函数在 TorchScript 中需要 dtype、layout、device 作为参数,但在 Python 中这些参数是可选的。
PyTorch 不支持的模块和类¶
TorchScript 目前无法编译许多其他常用的 PyTorch 结构。下面列出了 TorchScript 不支持的模块,以及不支持的 PyTorch 类的不完全列表。对于不支持的模块,我们建议使用 torch.jit.trace()
.
torch.autograd.enable_grad