TorchScript 不支持的 PyTorch 构造

不支持的张量方法

不支持的张量属性

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()