overshiki/kmeans_pytorch

pytorch implementation of basic kmeans algorithm(lloyd method with forgy initialization) with gpu support

语言：Python · GitHub 星标：94 · Fork：28

话题：gpu · kmeans · pytorch

在 GitHub 上查看原仓库