Python API

Overview

This API section details functions, modules, and objects included in Apache MXNet, describing what they are and what they do. The APIs are grouped into the following categories:

Imperative API

mxnet.ndarrayndarray/index.html

Imperative APIs to manipulate multi-dimensional arrays.

mxnet.gluongluon/index.html

Imperative APIs to load data, construct and train neural networks.

Gluon related modules

mxnet.autogradautograd/index.html

Functions for Automatic differentiation.

mxnet.optimizeroptimizer/index.html

Functions for applying an optimizer on weights.

mxnet.initializerinitializer/index.html

Default behaviors to initialize parameters.

mxnet.lr_schedulerlr_scheduler/index.html

Scheduling the learning rate.

mxnet.metricmetric/index.html

Metrics to evaluate the performance of a learned model.

mxnet.kvstorekvstore/index.html

Key value store interface of MXNet for parameter synchronization.

mxnet.contextmxnet/context/index.html

CPU and GPU context information.

mxnet.profilermxnet/profiler/index.html

Profiler setting methods.

mxnet.randommxnet/random/index.html

Imperative random distribution generator functions.

Symbolic API

mxnet.symsymbol/index.html

Symbolic APIs for multi-dimensional arrays and neural network layers

mxnet.modulemodule/index.html

Intermediate and high-level interface for performing computation with Symbols.

Symbol related modules

mxnet.callbackmxnet/callback/index.html

Functions to track various statuses during an epoch.

mxnet.monitormxnet/monitor/index.html

Outputs, weights, and gradients for debugging

mxnet.imagemxnet/image/index.html

Image iterators and image augmentation functions.

mxnet.iomxnet/io/index.html

Data iterators for common data formats and utility functions.

mxnet.recordiomxnet/recordio/index.html

Read and write for the RecordIO data format.

mxnet.visualizationmxnet/visualization/index.html

Functions for Symbol visualization.

Advanced modules

mxnet.executormxnet/executor/index.html

Managing symbolic graph execution.

mxnet.kvstore_servermxnet/kvstore_server/index.html

Server node for the key value store.

mxnet.enginemxnet/engine/index.html

Engine properties management.

mxnet.executor_managermxnet/executor_manager/index.html

Executor manager

mxnet.rtcmxnet/rtc/index.html

Tools for compiling and running CUDA code from the python frontend.

mxnet.test_utilsmxnet/test_utils/index.html

Tools for using and testing MXNet.

mxnet.utilmxnet/util/index.html

General utility functions

---

Did this page help you?

Yes

No

Thanks for your feedback!

---