paddle.utils 目录下包含飞桨框架工具类的 API。具体如下:
API 名称 | API 功能 |
---|---|
:ref:`load <cn_api_paddle_utils_cpp_extension_load>` |
飞桨框架一键编译自定义 OP、自动生成和返回 Python API 的接口 |
:ref:`setup <cn_api_paddle_utils_cpp_extension_setup>` |
飞桨框架编译自定义 OP、并安装到 site-package 目录的接口 |
:ref:`CppExtension <cn_api_paddle_utils_cpp_extension_CppExtension>` |
飞桨框架编译仅支持 CPU 的自定义 OP 扩展类 |
:ref:`CUDAExtension <cn_api_paddle_utils_cpp_extension_CUDAExtension>` |
飞桨框架编译支持 GPU 的自定义 OP 扩展类 |
:ref:`get_build_directory <cn_api_paddle_utils_cpp_extension_get_build_directory>` |
返回一键编译自定义 OP 的 build 目录 |