code: def __init__(self, in_dim, out_dim):
      super().__init__()
  
      self.weight = self.params.get(
          "weight_orig",
          shape=(out_dim, in_dim),
          grad_req="write",
          init=mx.init.Normal(1),
      )
      self.bias = self.params.get(
          "bias", shape=(out_dim), grad_req="write", init=mx.init.Zero()
      )
      self.num_hidden = out_dim
  
imports: import numpy
  from math import sqrt
  import mxnet
  import mxnet.ndarray
  import mxnet.gluon.nn
  from numpy import prod
label_desc: Use image node_id 1 to create a new EqualLinear object from inherited base classes: nn with example: obj = EqualLinear(in_dim, out_dim)