TorchScript 不支援的 PyTorch 建構

Torch 和 Tensor 不支援的屬性

TorchScript 支援在 torch 和 torch.Tensor 上定義的大多數方法，但我們沒有完全涵蓋。以下列出了一些已知的運算子和運算子類別，它們在 Python 和 TorchScript 之間的行為有所不同。如果您遇到其他不受支援的內容，請在 GitHub 上提出問題。以下未列出已棄用的運算子。

不支援的張量方法

不支援的張量屬性

在 Torch 上未正確繫結的函式

以下函式如果在 TorchScript 中使用將會失敗，因為它們沒有繫結在 torch 上，或者因為 Python 預期的結構描述與 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 之間結構描述不同的運算子

以下類別的運算子具有不同的結構描述

從非張量輸入建構張量的函式不支援 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