Contents:
Returns the index of the nearest centroid for a specific point
point (jax.numpy.ndarray) – A specific point
centroids (jax.numpy.ndarray) – An array of centroids
The index of the nearest centroid
(int)