Predicts the closest cluster for each item provided.
- Return type
Tensor of shape
[num_datapoints]with the index of the closest cluster for each datapoint.
When calling this function in a multi-process environment, each process receives only a subset of the predictions. If you want to aggregate predictions, make sure to gather the values returned from this method.