GOOD.utils.fast_pytorch_kmeans.init_methods