当前位置: 首页 > 文档资料 > PyTorch 中文文档 >

torch.utils.ffi

优质
小牛编辑
134浏览
2023-12-01
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 or List[str]) – 只包含导出函数的头文件列表
  • sources (List[str]) – 用于编译的sources列表
  • verbose (bool, optional) – 如果设置为False,则不会打印输出(默认值:True)。
  • with_cuda (bool, optional) – 设置为True以使用CUDA头文件进行编译(默认值:False)。
  • package (bool, optional) – 设置为True以在程序包模式下构建(对于要作为pip程序包安装的模块)(默认值:False)。
  • relative_to (str, optional) –构建文件的路径。packageTrue时需要。最好使用__file__作为参数。
  • kwargs – 传递给ffi以声明扩展的附加参数。有关详细信息,请参阅Extension API reference。