Printing the centroid sample for each label

[peonycabbage]

Hello, thank you very much for creating torch_kmeans. I can see that the centers, which are the corresponding coordinates of cluster centers, can be printed. I wonder how to print the centroid sample for each label.

[peonycabbage]

I also would like to ask if the order that the centers are printed follows the order of the clusters, for example, cluster 0's center is the first entry when we print the centers. Thank you in advance