Faiss对一些基础算法提供了非常高效的实现:k-means.PCA.PQ编解码. 聚类 假设2维tensor x: ncentroids = 1024 niter = 20 verbose = True d = x.shape[1] kmeans = faiss.Kmeans(d, ncentroids, niter, verbose) kmeans.train(x) 中心点放在kmeans.centroids中,目标函数的值放在kmeans.obj中.返回查询数据最近的中心点: D, I =…