original_docstring: Given that there exists some array arr where pts = arr[mask], this
      finds arr[mask2] where mask2 is some subset of mask.
  
      Args:
          pts (np.ndarray): arr[mask]
          mask (np.ndarray): some bool mask of the unknown array arr
          mask2 (np.ndarray): a subset of mask, i.e. a bool array that is false
              everywhere that mask is false but potentially false where mask is
              true
  
      Returns:
         arr[mask2]
      
docstring: Given that there exists some array arr where pts = arr[mask], this
  finds arr[mask2] where mask2 is some subset of mask.
docstring_tokens: Given that there exists some array arr where pts = arr [ mask ] this finds arr [ mask2 ] where mask2 is some subset of mask .
code: def submask(pts: np.ndarray, mask: np.ndarray, mask2: np.ndarray):
      """Given that there exists some array arr where pts = arr[mask], this
      finds arr[mask2] where mask2 is some subset of mask.
  
      Args:
          pts (np.ndarray): arr[mask]
          mask (np.ndarray): some bool mask of the unknown array arr
          mask2 (np.ndarray): a subset of mask, i.e. a bool array that is false
              everywhere that mask is false but potentially false where mask is
              true
  
      Returns:
         arr[mask2]
      """
      tus.check_ndarrays(
          mask=(mask, ('samples',), 'bool'),
          mask2=(mask2, (('samples', mask.shape[0]),), 'bool'))
      shape = [mask.shape[0]]
      shape.extend(pts.shape[1:])
      arr = np.zeros(shape, dtype=pts.dtype)
      arr[mask] = pts
      return arr[mask2]
code_tokens: def submask ( pts : np . ndarray , mask : np . ndarray , mask2 : np . ndarray ) : tus . check_ndarrays ( mask = ( mask , ( 'samples' , ) , 'bool' ) , mask2 = ( mask2 , ( ( 'samples' , mask . shape [ 0 ] ) , ) , 'bool' ) ) shape = [ mask . shape [ 0 ] ] shape . extend ( pts . shape [ 1 : ] ) arr = np . zeros ( shape , dtype = pts . dtype ) arr [ mask ] = pts return arr [ mask2 ]
short_docstring: Given that there exists some array arr where pts = arr[mask], this
  finds arr[mask2] where mask2 is some subset of mask.
short_docstring_tokens: Given that there exists some array arr where pts = arr [ mask ] this finds arr [ mask2 ] where mask2 is some subset of mask .
comment: """Given that there exists some array arr where pts = arr[mask], this
      finds arr[mask2] where mask2 is some subset of mask.
  
      Args:
          pts (np.ndarray): arr[mask]
          mask (np.ndarray): some bool mask of the unknown array arr
          mask2 (np.ndarray): a subset of mask, i.e. a bool array that is false
              everywhere that mask is false but potentially false where mask is
              true
  
      Returns:
         arr[mask2]
      """