[Help] Reaching definition of Aten matmul function

1

I’m figuring out where matmul function in Pytorch is and how it works.

Looked at this doc, I found matmul → __matmul_impl → at::mm_out, but I didn’t found any documentation for at::mm_out. The Pytorch repo is just too big to analyze. Is there any expert can explain how to find definition or read source code of Pytorch more efficient? Thank you.

2

have a look at this answer: How to find the c++/cuda implementation of specific operators in pytorch source code - #3 by bdhirsh - PyTorch Dev Discussions