Question about source code of pytorch

linyu (linyu) August 16, 2018, 1:48am 1

Where can I find the source code of torch.mm?

1 Like

SimonW (Simon Wang) August 16, 2018, 5:54am 2

It eventually dispatches to https://github.com/pytorch/pytorch/blob/2e0dd8690320fb1a7ecd548730824c1610207179/aten/src/ATen/native/LinearAlgebra.cpp#L136-L148, which calls blas gemm.

1 Like

linyu (linyu) August 16, 2018, 11:15am 3

Thank you for your apply!