Hi @glaringlee,
Thank you for clarifying this.
Separately from this topic, but somewhat related, the implementation for torch.distributions.Normal appears to be here: normal.py 9. In the sample() function, on line 63, there is a call to torch.normal(). I have not been able to find the implementation for it. Do you think you could provide a link to the file defining torch.normal()?