code: def maxpool2d_inshape(module_masks, mask):
      """
      Assume only the second dimension is masked
  
      Parameters
      ----------
      module_masks : ModuleMasks
          The ModuleMasks instance of the maxpool2d
      mask : CoarseMask
          The mask of its input tensor
  
      Returns
      -------
      CoarseMask
          The mask of its output tensor
      """
      assert isinstance(mask, CoarseMask)
      assert mask.mask_index[1] is not None
      assert mask.mask_index[0] is None
      assert mask.mask_index[2] is None
      assert mask.mask_index[3] is None
      if module_masks.input_mask is not None:
          assert module_masks.input_mask <= mask
      # assert module_masks.input_mask is None
      module_masks.set_input_mask(mask)
      module_masks.set_output_mask(mask)
      return mask
  
imports: import logging
  import torch
label_desc: Use image node_id 15 for calling a global function with example usage: maxpool2d_inshape(module_masks, mask) and returns: mask