Balanced KMeans: number of sizes provided must be greater or equal to the number of dimensions in the tensor

Question

I\'m been trying to make a balanced clusters for my masther thesis using Google Colab (because I don\'t have a GPU) and balanced k-means Library (https://pypi.org/project/ba