How to get batch of clusters

[Boltzmachine]

For example, if you use functions like fps or voxel_grid,

cluster = voxel_grid(x, size=size, batch=batch)

where cluster is the assignment of each x. My question is, how do I get the batch of each cluster index?