How to sample unique data?

samples=torch.multinomial(probs,5)

but samples data may be duplicate

TLDR: use WeightedRandomSampler with replace=False