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) –构建文件的路径。 package为True时需要。最好使用__file__作为参数。
  8. kwargs – 传递给ffi以声明扩展名的其他参数。详情请参阅扩展API参考。

译者署名

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

results matching ""

    No results matching ""