source
espnet.utils.spec_augment.flatten_grid_locations(grid_locations, image_height, image_width)
Flatten grid locations.