source
espnet.utils.spec_augment.get_flat_grid_locations(image_height, image_width, device)
Get Flat Grid Locations.