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