code: def get_weight_key(module):
      for k in module.params.keys():
          if "weight" in k:
              weight_key = k
  
      return weight_key
  
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 for calling a global function with example usage: get_weight_key(module) and returns: weight_key