search

subhadarship / kmeans_pytorch

kmeans using PyTorch
https://subhadarship.github.io/kmeans_pytorch
MIT License
470 stars 75 forks source link

Can input like is (batch,node,feature)? #33

Open 280777510 opened 1 year ago

280777510 commented 1 year ago

Hello, GPU-accelerated version of Kmeans, can pass in a tensor whose shape is 3 dimensional? For example, (3,758,32) is (batch,node,feature), and the batch is expected to be parallel.