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
  ----------
  mask : CoarseMask
      The mask for output
docstr_tok: ["Parameters","--","--","--","--","--","mask",":","CoarseMask","The","mask","for","output"]
code: def set_output_mask(self, mask):
      """
      Parameters
      ----------
      mask : CoarseMask
          The mask for output
      """
      self.output_mask = mask
  
imports: import logging
  import torch
label_desc: Use image node_id 4 for calling the ModuleMasks obj's underlying member method code with example usage: obj.set_output_mask(mask) without return types