Raw source code taken from:
https://github.com/scikit-learn/scikit-learn/blob/579e7de7f38f9f514ff2b2be049e67b14e723d17/sklearn/cluster/_kmeans.py#L525

Not functional, just for sake of reference