LeviViana/torch_sampling

Efficient reservoir sampling implementation for PyTorch

Cuda