Mini-batch k-means clustering for PyTorch tensors

Mini-batch k-means 是一个针对 PyTorch 张量的 mini-batch k-means 聚类库，遵循 Sculley 的 Web-Scale K-Means Clustering 算法。该库支持 CPU 和 GPU，可直接在 PyTorch 张量上运行，并提供模型保存和加载功能。