torch.matmul在哪里实现了，特别是在GPU上运行的部分？
整个项目有2M行代码。我尝试grep 1.8.2的来源发布，但找不到此功能。我猜它是由某些东西生成的...
最佳答案
应该执行如下: https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/native/LinearAlgebra.cpp#L1450
matmul Functions.h 包含的功能和 ATen.h包括 Functions.h
在 pytorch api( pytorch/torch/csrc/api/include/torch/linalg.h ) 中，它包括 <ATen/ATen.h>
所以:
pytorch/torch/csrc/api/include/torch/linalg.h > ATen > 函数 > 线性代数 > matmul
关于python - torch.matmul 在哪里实现？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/69264554/