TorchScript 不支援的 PyTorch 構造

Torch 和 Tensor 不支援的屬性

TorchScript 支援在 torch 和 torch.Tensor 上定義的大多數方法，但我們尚未完全覆蓋。以下是已知的一些特定運算元和運算元類別，它們在 Python 和 TorchScript 之間的行為有所差異。如果您遇到其他不受支援的情況，請提交 GitHub Issue。已棄用的運算元未在此處列出。

不支援的 Tensor 方法

不支援的 Tensor 屬性

在 Torch 上未正確繫結的函式

以下函式如果在 TorchScript 中使用將會失敗，原因可能是在 torch 上未繫結它們，或者 Python 期望的 schema 與 TorchScript 不同。

• torch.tensordot()

• torch.nn.init.calculate_gain()

• torch.nn.init.eye_()

• torch.nn.init.dirac_()

• torch.nn.init.kaiming_normal_()

• torch.nn.init.orthogonal_()

• torch.nn.init.sparse()

Torch 與 Python 之間 Schema 存在差異的運算元

以下類別的運算元具有不同的 schema

從非 tensor 輸入構造 tensor 的函式不支援 requires_grad 引數，torch.tensor 除外。這包括以下運算元

• torch.norm()

• torch.bartlett_window()

• torch.blackman_window()

• torch.empty()

• torch.empty_like()

• torch.empty_strided()

• torch.eye()

• torch.full()

• torch.full_like()

• torch.hamming_window()

• torch.hann_window()

• torch.linspace()

• torch.logspace()

• torch.normal()

• torch.ones()

• torch.rand()

• torch.rand_like()

• torch.randint_like()

• torch.randn()

• torch.randn_like()

• torch.randperm()

• torch.tril_indices()

• torch.triu_indices()

• torch.vander()

• torch.zeros()

• torch.zeros_like()

以下函式在 TorchScript 中需要 dtype、layout、device 作為引數，但在 Python 中這些引數是可選的。

• torch.randint()

• torch.sparse_coo_tensor()

• to()

PyTorch 不支援的模組和類

TorchScript 目前無法編譯許多其他常用的 PyTorch 構造。下面列出了 TorchScript 不支援的模組以及一個不完整的 PyTorch 類列表。對於不支援的模組，我們建議使用 torch.jit.trace()。

• torch.nn.RNN

• torch.nn.AdaptiveLogSoftmaxWithLoss

• torch.autograd.Function

• torch.autograd.enable_grad