osc.viz.backbone.kmeans_clusters

kmeans_clusters(x, n_clusters=11)

Batched K-means clustering.

Parameters

• x (Union[Tensor, ndarray]) – N samples of C-dimensional features with leading batch dimensions, e.g. shape […, N, C]

• n_clusters (int) – desired number of clusters

Return type

ndarray

Returns

int array of cluster IDs, shape […, N, C]