code: def mask_pts_at_padded_regions(
      grid_pt: Tensor, mask: Tensor
  ) -> Tensor:
      """For megadepth dataset, zero-padding exists in images."""
      n, h, w = mask.shape
      mask = mask.reshape(n, h * w).unsqueeze(-1).repeat(1, 1, 2)
      # from einops import repeat
      # mask = repeat(mask, 'n h w -> n (h w) c', c=2)
      grid_pt[~mask.bool()] = 0
      return grid_pt
  
file_path: kornia/kornia/feature/loftr/utils/supervision.py
imports: from __future__ import annotations
  from typing import Any
  import torch
  from kornia.core import Tensor
  from kornia.utils import create_meshgrid
  from .geometry import warp_kpts
label_desc: Use image node_id 1 for calling a global function with example usage: mask_pts_at_padded_regions(grid_pt, mask) and returns: grid_pt