ott.neural.models.ICNN.gaussian_map_samples