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

[Jayati-Kaushik]

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

[DecaiJin]

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