search

subhadarship / kmeans_pytorch

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

IndexError: index_select(): Index is supposed to be a vector #34

Open Jayati-Kaushik opened 1 year ago

Jayati-Kaushik commented 1 year ago

This error occurs when I try to cluster with a set of tensors with n_clusters = 2.

DecaiJin commented 1 year ago

It may be not support the batched input. You can test the script with the shape (n_samples, features)