API 文档

MindSpore提供了丰富的模型构建、训练、推理等接口,各模块接口功能说明如下。

模块名

模块说明

mindspore

框架基础接口。

mindspore.nn

神经网络层,用于构建神经网络中的预定义构建块或计算单元。

mindspore.ops

函数接口。

mindspore.ops.primitive

Primitive的算子。

mindspore.mint

与业界主流用法一致的functional、nn、优化器接口。

mindspore.amp

混合精度接口。

mindspore.train

训练接口。

mindspore.communication

集合通信接口。

mindspore.communication.comm_func

集合通信函数式接口。

mindspore.common.initializer

参数初始化。

mindspore.hal

设备管理、流管理、事件管理与内存管理的接口。

mindspore.dataset

加载和处理各种数据集的接口。

mindspore.dataset.transforms

通用数据变换。

mindspore.mindrecord

MindSpore开发的高效数据格式MindRecord相关的操作接口。

mindspore.nn.probability

可参数化的概率分布和采样函数。

mindspore.rewrite

基于自定义规则的模型源码修改接口。

mindspore.multiprocessing

多进程接口。

mindspore.boost

自动加速网络接口。

mindspore.numpy

类NumPy接口。

mindspore.scipy

类SciPy接口。

mindspore.utils

工具接口。

mindspore.experimental

实验性接口。

环境变量

环境变量相关说明。