Function find_closest_centroid

fn find_closest_centroid(
    sub_vec: &[f32],
    centroids: &[f32],
    sub: usize,
    k: usize,
    sub_dim: usize,
) -> usize