@cruzas
That’s a binding function which actually pointing to our c++ implementation, entry point is here:
‘https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/native/Distributions.cpp#L286 57’
The cpu implementation is here:
https://github.com/pytorch/pytorch/blob/5ab36ec98ba28b61ac3be57427d67ad8e1f0bab4/aten/src/ATen/native/cpu/DistributionTemplates.h#L173 26
The gpu implementation is here:
https://github.com/pytorch/pytorch/blob/34523b70c1369024f805a0480bc39a5f21e64a03/aten/src/ATen/native/cuda/DistributionTemplates.h#L407 22