Python PyTorch CppExtension用法及代碼示例

本文簡要介紹python語言中 torch.utils.cpp_extension.CppExtension 的用法。

用法:

torch.utils.cpp_extension.CppExtension(name, sources, *args, **kwargs)

為 C++ 創建一個 setuptools.Extension。

創建setuptools.Extension 的便捷方法，使用最少(但通常足夠)的參數來構建 C++ 擴展。

所有參數都轉發到setuptools.Extension 構造函數。

示例

>>> from setuptools import setup
>>> from torch.utils.cpp_extension import BuildExtension, CppExtension
>>> setup(
        name='extension',
        ext_modules=[
            CppExtension(
                name='extension',
                sources=['extension.cpp'],
                extra_compile_args=['-g']),
        ],
        cmdclass={
            'build_ext': BuildExtension
        })