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::ones
或 at::randn
或 at::
命名空间中的任何其他张量工厂函数创建张量时出现错误。
修复:将工厂函数调用中的 at::
替换为 torch::
。您永远不应该使用 at::
命名空间中的工厂函数,因为它们会创建张量。相应的 torch::
函数会创建变量,而您的代码中应该只处理变量。