espnet.utils.spec_augment.get_grid_locations

source

espnet.utils.spec_augment.get_grid_locations(image_height, image_width, device)

Get grid locations.