API
MindSpore provides rich interfaces for model building, training, and inference. The functions of each module interface are described below.
Module Name |
Descriptions |
---|---|
Framework foundation interface. |
|
A neural network layer for building predefined building blocks or computational units in a neural network. |
|
Function interface. |
|
Primitive operator. |
|
Functional, nn, and optimizer interfaces consistent with mainstream industry usage. |
|
Mixed-precision interface. |
|
Traning interface. |
|
Collection communication interface. |
|
Collection communication functional interface. |
|
Parameter initialization. |
|
Interface of device management, stream management, event management and memory management. |
|
Interfaces for loading and processing various datasets. |
|
Generalized data transformations. |
|
Operations interface related to efficient data format MindRecord developed by MindSpore. |
|
Parameterizable probability distributions and sampling functions. |
|
Custom rule-based model source code modification interface. |
|
Multi-processing interface. |
|
Automatic acceleration network interfaces. |
|
NumPy Class interface. |
|
SciPy Class interface. |
|
Tools interface. |
|
Experimental interface. |
|
Notes related to environment variables. |