API 文档
MindSpore提供了丰富的模型构建、训练、推理等接口,各模块接口功能说明如下。
模块名 |
模块说明 |
---|---|
框架基础接口。 |
|
神经网络层,用于构建神经网络中的预定义构建块或计算单元。 |
|
函数接口。 |
|
Primitive的算子。 |
|
与业界主流用法一致的functional、nn、优化器接口。 |
|
混合精度接口。 |
|
训练接口。 |
|
集合通信接口。 |
|
集合通信函数式接口。 |
|
参数初始化。 |
|
设备管理、流管理、事件管理与内存管理的接口。 |
|
加载和处理各种数据集的接口。 |
|
通用数据变换。 |
|
MindSpore开发的高效数据格式MindRecord相关的操作接口。 |
|
可参数化的概率分布和采样函数。 |
|
基于自定义规则的模型源码修改接口。 |
|
多进程接口。 |
|
自动加速网络接口。 |
|
类NumPy接口。 |
|
类SciPy接口。 |
|
工具接口。 |
|
实验性接口。 |
|
环境变量相关说明。 |