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