gluon.nn

Gluon provides a large number of build-in neural network layers in the following two modules:

mxnet.gluon.nn

mxnet.gluon.contrib.nn

We group all layers in these two modules according to their categories.

Sequential containers

nn.Sequential

nn.HybridSequential

Basic Layers

nn.Dense

nn.Activation

nn.Dropout

nn.Flatten

nn.Lambda

nn.HybridLambda

Convolutional Layers

nn.Conv1D

nn.Conv2D

nn.Conv3D

nn.Conv1DTranspose

nn.Conv2DTranspose

nn.Conv3DTranspose

Pooling Layers

nn.MaxPool1D

nn.MaxPool2D

nn.MaxPool3D

nn.AvgPool1D

nn.AvgPool2D

nn.AvgPool3D

nn.GlobalMaxPool1D

nn.GlobalMaxPool2D

nn.GlobalMaxPool3D

nn.GlobalAvgPool1D

nn.GlobalAvgPool2D

nn.GlobalAvgPool3D

nn.ReflectionPad2D

Normalization Layers

nn.BatchNorm

nn.InstanceNorm

nn.LayerNorm

Embedding Layers

nn.Embedding

Advanced Activation Layers

nn.LeakyReLU

nn.PReLU

nn.ELU

nn.SELU

nn.Swish

API Reference