评价此页

FAQ#

以下列出了用户在使用 C++ API 的各个部分时遇到的若干常见问题。

C++ 扩展#

来自 PyTorch/ATen 的未定义符号错误#

问题:导入扩展时出现 ImportError,提示 PyTorch 或 ATen 的某些 C++ 符号未定义。例如

>>> import extension
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: /home/user/.pyenv/versions/3.7.1/lib/python3.7/site-packages/extension.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN2at19UndefinedTensorImpl10_singletonE

修复:在导入扩展之前,通过 import torch 来修复。这将使您的扩展所依赖的 PyTorch 动态(共享)库中的符号可用,从而允许在导入扩展时解析它们。

使用 at:: 中的函数创建张量时出错#

问题:使用例如 at::onesat::randnat:: 命名空间中的任何其他张量工厂函数创建张量时出现错误。

修复:将工厂函数调用中的 at:: 替换为 torch::。您永远不应该使用 at:: 命名空间中的工厂函数,因为它们会创建张量。相应的 torch:: 函数会创建变量,而您的代码中应该只处理变量。