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

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

环境变量

环境变量相关说明。

PyTorch与MindSpore API映射表

PyTorch与MindSpore的API映射关系和差异分析。