文件名称：kmeans_pytorch：使用PyTorch的kmeans
文件大小：961KB
文件格式：ZIP
更新时间：2024-02-24 21:22:10
jekyll github-pages docs gpu pytorch
使用PyTorch的K均值 kmeans的PyTorch实现以利用GPU 入门 import torch import numpy as np from kmeans_pytorch import kmeans # data data_size, dims, num_clusters = 1000, 2, 3 x = np.random.randn(data_size, dims) / 6 x = torch.from_numpy(x) # kmeans cluster_ids_x, cluster_centers = kmeans( X=x, num_clusters=num_clu
【文件预览】：
kmeans_pytorch-master
----setup.py(2KB)
----batch_processing.ipynb(136KB)
----.gitignore(205B)
----MANIFEST.in(117B)
----LICENSE(1KB)
----cpu_vs_gpu.ipynb(59KB)
----README.md(2KB)
----tests()
--------test_project.py(279B)
----makefile(270B)
----kmeans_pytorch()
--------__init__.py(5KB)
--------main.py(138B)
----example.ipynb(48KB)
----docs()
--------index.md(1KB)
--------chapters()
--------static()
--------_config.yml(829B)
--------_layouts()
--------404.html(260B)