Example of creating a network using nn.Block class

mxnet_gluon_lenet_nnBlock.py

from mxnet import nn
class myLeNet(nn.Block):
    def __init__(self, **kwargs):
        super(myLeNet, self).__init__(**kwargs)
        self.conv1 = nn.Conv2D(channels=6,kernel_size=5,activation='relu')
        self.pool1 = nn.MaxPool2D(pool_size=2,strides=2)
        self.conv2 = nn.Conv2D(channels=16,kernel_size=3,activation='relu')
        self.pool1 = nn.MaxPool2D(pool_size=2,strides=2)
        self.fc1 = nn.Dense(120, activation='relu')
        self.fc2 = nn.Dense(84, activation='relu')
        self.fc3 = nn.Dense(10)
        
    def forward(self,x):
        return self.fc3(self.fc2(self.fc1(self.pool1(self.conv2(self.pool1(self.conv1(x)))))))