code: def __init__(self, channel):
      super().__init__()
  
      self.weight = self.params.get(
          "weight_orig", shape=(1, channel, 1, 1), init=mx.init.Zero()
      )
  
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 NoiseInjection object from inherited base classes: nn with example: obj = NoiseInjection(channel)