I assume you have input data and a clustering algorithm, that returns the corresponding cluster for each sample.
If that’s the case, you can simply split the data using this code:
data = torch.randn(100, 2)
clusters = torch.empty(100, dtype=torch.long).random_(2)

data_class0 = data[clusters==0]
data_class1 = data[clusters==1]