code: def compute_weight(weight_orig):
      fan_in = weight_orig.shape[1] * weight_orig[0][0].size
  
      return weight_orig * sqrt(2 / (fan_in + 1e-8))
  
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 2 for calling a global function with example usage: compute_weight(weight_orig) and returns: unknown