Lab6: Sampling and PyTorch