cr.sparse.cluster.vq.find_nearest_jit

cr.sparse.cluster.vq.find_nearest_jit(point, centroids)

Returns the index of the nearest centroid for a specific point

Parameters

• point (jax.numpy.ndarray) – A specific point

• centroids (jax.numpy.ndarray) – An array of centroids

Returns

The index of the nearest centroid

Return type

(int)