class_docstr: The masks of a module, including the masks for weights, inputs, output
class_docstr_tok: ["The","masks","of","a","module",",","including","the","masks","for","weights",",","inputs",",","output"]
docstr: Parameters
  ----------
  module_name : str
      The name of the module or function
docstr_tok: ["Parameters","--","--","--","--","--","module_name",":","str","The","name","of","the","module","or","function"]
code: def __init__(self, module_name, module=None):
      """
      Parameters
      ----------
      module_name : str
          The name of the module or function
      """
      self.module_name = module_name
      self.module = module
      self.param_masks = dict()
      self.input_mask = None
      self.output_mask = None
  
imports: import logging
  import torch
label_desc: Use image node_id 1 to create a new ModuleMasks object  with example: obj = ModuleMasks(module_name, module)