How to get a uniform distribution in a range [r1,r2] in PyTorch? How to get a uniform distribution in a range [r1,r2] in PyTorch? python python