torch.utils.ffi

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

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

参数：

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

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