Skip to content

Latest commit

 

History

History
72 lines (53 loc) · 3.79 KB

File metadata and controls

72 lines (53 loc) · 3.79 KB

paddle.utils

paddle.utils 目录下包含飞桨框架工具类的 API。具体如下:

自定义 OP 相关 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 目录

工具类相关 API

API 名称 API 功能
:ref:`deprecated <cn_api_paddle_utils_deprecated>`
飞桨框架废弃 API 装饰器
:ref:`get_weights_path_from_url <cn_api_paddle_utils_download_get_weights_path_from_url>`
从文件夹获取权重
:ref:`run_check <cn_api_paddle_utils_run_check>`
检查是否正常安装飞桨框架
:ref:`generate <cn_api_paddle_utils_unique_name_generate>`
产生以前缀开头的唯一名称
:ref:`guard <cn_api_paddle_utils_unique_name_guard>`
更改命名空间
:ref:`switch <cn_api_paddle_utils_unique_name_switch>`
切换命名空间
:ref:`Profiler <cn_api_paddle_profiler_Profiler>`
通用性能分析器
:ref:`require_version <cn_api_paddle_utils_require_version>`
用于检查已安装的飞桨版本是否介于[min_version, max_version]之间
:ref:`to_dlpack <cn_api_paddle_to_dlpack>`
用于将 Tensor 对象转换为 DLPack
:ref:`from_dlpack <cn_api_paddle_from_dlpack>`
用于从 DLPack 中解码出 Tensor 对象
:ref:`try_import <cn_api_paddle_utils_try_import>`
用于尝试导入一个模块并且在失败时提供自定义错误信息

别名相关 API

API 名称 API 功能
:ref:`default_collate <cn_api_paddle_utils_data_default_collate>`
paddle.io.dataloader.collate.default_collate_fn 的别名
:ref:`get_worker_info <cn_api_paddle_utils_data_get_worker_info>`
paddle.io.get_worker_info 的别名
:ref:`ChainDataset <cn_api_paddle_utils_data_ChainDataset>`
paddle.io.ChainDataset 的别名
:ref:`ConcatDataset <cn_api_paddle_utils_data_ConcatDataset>`
paddle.io.ConcatDataset 的别名
:ref:`Dataset <cn_api_paddle_utils_data_Dataset>`
paddle.io.Dataset 的别名
:ref:`IterableDataset <cn_api_paddle_utils_data_IterableDataset>`
paddle.io.IterableDataset 的别名
:ref:`Subset <cn_api_paddle_utils_data_Subset>`
paddle.io.Subset 的别名
:ref:`random_split <cn_api_paddle_utils_data_random_split>`
paddle.io.random_split 的别名
:ref:`BatchSampler <cn_api_paddle_utils_data_BatchSampler>`
paddle.io.BatchSampler 的别名
:ref:`RandomSampler <cn_api_paddle_utils_data_RandomSampler>`
paddle.io.RandomSampler 的别名
:ref:`Sampler <cn_api_paddle_utils_data_Sampler>`
paddle.io.Sampler 的别名
:ref:`SequentialSampler <cn_api_paddle_utils_data_SequentialSampler>`
paddle.io.SequenceSampler 的别名