code: def calc_updown(self, orig_weight):
      w = self.w.to(orig_weight.device, dtype=orig_weight.dtype)
  
      output_shape = [w.size(0), orig_weight.size(1)]
      if self.on_input:
          output_shape.reverse()
      else:
          w = w.reshape(-1, 1)
  
      updown = orig_weight * w
  
      return self.finalize_updown(updown, orig_weight, output_shape)
  
imports: import network
label_desc: Use image node_id 2 for calling the NetworkModuleIa3 obj's underlying member method code with example usage: obj.calc_updown(orig_weight) and returns: self