API 文档

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

模块名

模块说明

mindspore

框架基础接口。

mindspore.ops

函数接口。

mindspore.nn

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

mindspore.mint

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

mindspore.common.initializer

参数初始化。

mindspore.amp

混合精度接口。

mindspore.train

训练接口。

mindspore.parallel

自动并行接口。

mindspore.runtime

运行时接口。

mindspore.device_context

设备及后端管理接口。

mindspore.communication

集合通信接口。

mindspore.dataset

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

mindspore.numpy

类NumPy接口。

mindspore.scipy

类SciPy接口。

mindspore.multiprocessing

多进程接口。

mindspore.utils

工具接口。

mindspore.experimental

实验性接口。

mindspore.ops.primitive

Primitive的算子。

mindspore.boost

自动加速网络接口。

mindspore.nn.probability

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

mindspore.rewrite

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

mindspore.hal

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

环境变量

环境变量相关说明。