I need to sample from a normal distribution in the training process, and is there any way to generate random numbers from device, rather than using torch.normal(mean, variance).cuda(). As when the dimension of normal is large, it seems to be very costly. Thanks!