代码拉取完成,页面将自动刷新
译者:@之茗
校对者:@aleczhang
torch.utils.ffi.create_extension(name, headers, sources, verbose=True, with_cuda=False, package=False, relative_to='.', **kwargs)
创建并配置一个 cffi.FFI 对象, 用于构建 PyTorch 的扩展.
参数:
name (str)
– 包名. 可以是嵌套模块, 例如. .ext.my_lib
.headers (str 或 List[str])
– 只包含导出函数的头文件列表.sources (List[str])
– 用于编译的sources列表.verbose (bool, 可选)
– 如果设置为 False
, 则不会打印输出 (默认值: True).with_cuda (bool, 可选)
– 设置为 True
以使用 CUDA 头文件进行编译 (默认值: False)package (bool, 可选)
– 设置为 True
以在包模式下构建 (对于要作为 pip 程序包安装的模块) (默认值: False).relative_to (str, 可选)
– 构建文件的路径. 当 package 为 True
时需要. 最好使用 __file__
作为参数.kwargs
– 传递给 ffi 以声明扩展的附件参数. 参考 Extension API reference 查阅更详细内容.此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。