mindspore
Tensor
Tensor is a data structure that stores an n-dimensional array. |
|
A sparse representation of a set of nonzero elements from a tensor at given indices. |
|
Constructs a sparse tensor in CSR (Compressed Sparse Row) format, with specified values indicated by values and row and column positions indicated by indptr and indices. |
|
A sparse representation of a set of tensor slices at given indices. |
|
A sparse representation of a set of nonzero elements from a tensor at given indices. |
Parameter
Parameter is a Tensor subclass, when they are assigned as Cell attributes they are automatically added to the list of its parameters, and will appear, e.g. |
|
Inherited from tuple, ParameterTuple is used to save multiple parameter. |
DataType
- class mindspore.dtype
Create a data type object of MindSpore.
The actual path of
dtype
is/mindspore/common/dtype.py
. Run the following command to import the package:from mindspore import dtype as mstype
Numeric Type
Currently, MindSpore supports
Int
type,Uint
type,Float
type andComplex
type. The following table lists the details.Definition
Description
mindspore.int8
,mindspore.byte
8-bit integer
mindspore.int16
,mindspore.short
16-bit integer
mindspore.int32
,mindspore.intc
32-bit integer
mindspore.int64
,mindspore.intp
64-bit integer
mindspore.uint8
,mindspore.ubyte
unsigned 8-bit integer
mindspore.uint16
,mindspore.ushort
unsigned 16-bit integer
mindspore.uint32
,mindspore.uintc
unsigned 32-bit integer
mindspore.uint64
,mindspore.uintp
unsigned 64-bit integer
mindspore.float16
,mindspore.half
16-bit floating-point number
mindspore.float32
,mindspore.single
32-bit floating-point number
mindspore.float64
,mindspore.double
64-bit floating-point number
mindspore.complex64
64-bit complex number
mindspore.complex128
128-bit complex number
Other Type
For other defined types, see the following table.
Type
Description
tensor
MindSpore’s
tensor
type. Data format uses NCHW. For details, see tensor.bool_
Boolean
True
orFalse
.int_
Integer scalar.
uint
Unsigned integer scalar.
float_
Floating-point scalar.
complex
Complex scalar.
number
Number, including
int_
,uint
,float_
,complex
andbool_
.list_
List constructed by
tensor
, such asList[T0,T1,...,Tn]
, where the elementTi
can be of different types.tuple_
Tuple constructed by
tensor
, such asTuple[T0,T1,...,Tn]
, where the elementTi
can be of different types.function
Function. Return in two ways, when function is not None, returns Func directly, the other returns Func(args: List[T0,T1,…,Tn], retval: T) when function is None.
type_type
Type definition of type.
type_none
No matching return type, corresponding to the
type(None)
in Python.symbolic_key
The value of a variable is used as a key of the variable in
env_type
.env_type
Used to store the gradient of the free variable of a function, where the key is the
symbolic_key
of the free variable’s node and the value is the gradient.Tree Topology
The relationships of the above types are as follows:
└─────── number │ ├─── bool_ │ ├─── int_ │ │ ├─── int8, byte │ │ ├─── int16, short │ │ ├─── int32, intc │ │ └─── int64, intp │ ├─── uint │ │ ├─── uint8, ubyte │ │ ├─── uint16, ushort │ │ ├─── uint32, uintc │ │ └─── uint64, uintp │ ├─── float_ │ │ ├─── float16 │ │ ├─── float32 │ │ └─── float64 │ └─── complex │ ├─── complex64 │ └─── complex128 ├─── tensor │ ├─── Array[Float32] │ └─── ... ├─── list_ │ ├─── List[Int32,Float32] │ └─── ... ├─── tuple_ │ ├─── Tuple[Int32,Float32] │ └─── ... ├─── function │ ├─── Func │ ├─── Func[(Int32, Float32), Int32] │ └─── ... ├─── type_type ├─── type_none ├─── symbolic_key └─── env_type
Convert MindSpore dtype to numpy data type. |
|
Determine whether type_ is a subclass of dtype. |
|
Convert MindSpore dtype to python data type. |
|
Convert python type to MindSpore type. |
|
Get the MindSpore data type, which corresponds to python type or variable. |
Seed
Set global seed. |
|
Get global seed. |
Context
Set context for running environment. |
|
Get context attribute value according to the input key. |
|
Set auto parallel context, only data parallel supported on CPU. |
|
Get auto parallel context attribute value according to the key. |
|
Reset auto parallel context attributes to the default values: |
|
Parallel mode options. |
|
Set parameter server training mode context. |
|
Get parameter server training mode context attribute value according to the key. |
|
Reset parameter server training mode context attributes to the default values: |
|
Set parameters in the algorithm for parallel strategy searching. |
|
Get the algorithm parameter config attributes. |
|
Reset the algorithm parameter attributes. |
Model
High-Level API for training or inference. |
Callback
Abstract base class used to build a Callback class. |
|
The configuration of model checkpoint. |
|
Stop training when a monitored metric has stopped improving. |
|
Records the network outputs and metrics information into a History object. |
|
Callback for creating simple, custom callbacks. |
|
Change the learning_rate during training. |
|
Monitor the loss in train or monitor the loss and eval metrics in fit. |
|
The checkpoint callback class. |
|
Reduce learning rate when the monitor has stopped improving. |
|
Hold and manage information about the model. |
|
Monitor the time in train or eval process. |
Dataset Helper
DatasetHelper is a class to process the MindData dataset and provides the information of dataset. |
|
Connect the network with dataset in dataset_helper. |
|
A wrapper function to generate a function for the input function. |
Serialization
Get the status of asynchronous save checkpoint thread. |
|
Build strategy of every parameter in network. |
|
Convert mindir model to other format model. |
|
Export the MindSpore network into an offline model in the specified format. |
|
Load MindIR. |
|
Load checkpoint info from a specified file. |
|
Load checkpoint into net for distributed predication. |
|
Load parameters into network, return parameter list that are not loaded in the network. |
|
Merge parameter slices into one parameter. |
|
Parse data file generated by mindspore.ops.Print. |
|
Build rank list, the checkpoint of ranks in the rank list has the same contents with the local rank who saves the group_info_file_name. |
|
Save checkpoint to a specified file. |
JIT
Jit config for compile. |
|
Create a callable MindSpore graph from a Python function. |
|
Class decorator for user-defined classes. |
|
Make a constant value mutable. |
Log
Get the logger level. |
|
Get logger configurations. |
Installation Verification
Provide a convenient API to check if the installation is successful or failed. |
Debugging and Tuning
This class to enable the profiling of MindSpore neural networks. |
|
SummaryCollector can help you to collect some common information. |
|
SummaryLandscape can help you to collect loss landscape information. |
|
SummaryRecord is used to record the summary data and lineage data. |
|
Enable or disable dump for the target and its contents. |
Memory Recycle
Recycle memory used by MindSpore. |
Thor
Convert model to thor model. |
|
Convert net to thor layer net, used to compute and store second-order information matrix. |