torch.utils.ffi

torch.utils.ffi.create_extension(name, headers, sources, verbose=True, with_cuda=False, package=False, relative_to='.', **kwargs)

创建并配置一个构建PyTorch扩展的cffi.FFI对象。

参数:

  1. name (str) – 包名。可以是嵌套的模块如 .ext.my_lib
  2. headers (str or List[str]) – 仅包含导出函数的头文件列表
  3. sources (List[str]) – 要编译的源列表。
  4. verbose (bool, optional) – 如果设置为False,则不打印任何输出(默认值:True)。
  5. with_cuda (bool, optional) – 设置为True表示使用CUDA头进行编译(默认值:False)。
  6. package (bool, optional) – 设置为True表示以程序包的模式构建(对于要以pip包安装的模块)(默认值:False)。
  7. relative_to (str, optional) –构建文件的路径。 packageTrue时需要。最好使用__file__作为参数。
  8. kwargs – 传递给ffi以声明扩展名的其他参数。详情请参阅扩展API参考

译者署名

用户名 头像 职能 签名
Song 翻译 人生总要追求点什么