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		翻译	人生总要追求点什么