How to find source code of diag in torch

1

Maybe someone can teach me how to search the source code of diag in torch

2

Which function exactly?

For torch.diag, it maybe this: ATen/aten/src/TH/generic/THTensorMath.h at 4aa3e1de29ed58457e530f84217e53db0998476c · zdevito/ATen · GitHub