code: def get_grid_locations(image_height, image_width, device):
      y_range = torch.linspace(
          0, image_height - 1, image_height, device=device
      )
      x_range = torch.linspace(
          0, image_width - 1, image_width, device=device
      )
      y_grid, x_grid = torch.meshgrid(y_range, x_range)
      return torch.stack((y_grid, x_grid), -1)
  
file_path: espnet/espnet/utils/spec_augment.py
imports: import random
  import torch
label_desc: Use image node_id 6 for calling a global function with example usage: get_grid_locations(image_height, image_width, device) and returns: torch