class_docstr: MLP with linear output
class_docstr_tok: ["MLP","with","linear","output"]
code: def forward(self, x):
      if self.linear_or_not:
          return self.linear(x)
      else:
          h = x
          for i in range(self.num_layers - 1):
              h = nd.relu(self.batch_norms[i](self.linears[i](h)))
          return self.linears[-1](h)
  
file_path: dgl/examples/mxnet/gin/gin.py
imports: import mxnet
  from dgl.nn.mxnet.conv import GINConv
  from dgl.nn.mxnet.glob import AvgPooling, MaxPooling, SumPooling
  from mxnet import gluon, nd
  from mxnet.gluon import nn
label_desc: Use image node_id 2 for calling the MLP obj's underlying member method code with example usage: obj.forward(x) and returns: self, self