Pytorch 如何找到torch.solve的源代码?

Pytorch 如何找到torch.solve的源代码?,pytorch,linear-algebra,implementation,Pytorch,Linear Algebra,Implementation,我关心的是,torch.solve()是否检查线性系统的系数矩阵的条件并采用适当的预条件;因此,我对其实现细节感到好奇。我已经通读了好几个答案,试图找到源文件,但徒劳无功。我希望有人能帮我在ATen库中找到它的定义。我认为它只是使用LAPACK作为CPU,CUBLAS作为GPU,因为torch.solve列在屏幕上的“BLAS和LAPACK操作”下 然后我们寻找包装器代码,我相信是这样。这意味着它只是遵从了CPU版本的LAPACK和GPU版本的CUBLAS:我认为这就是包装器代码:@Multih

我关心的是,
torch.solve()
是否检查线性系统的系数矩阵的条件并采用适当的预条件;因此,我对其实现细节感到好奇。我已经通读了好几个答案,试图找到源文件,但徒劳无功。我希望有人能帮我在ATen库中找到它的定义。

我认为它只是使用LAPACK作为CPU,CUBLAS作为GPU,因为
torch.solve
列在屏幕上的“BLAS和LAPACK操作”下


然后我们寻找包装器代码,我相信是这样。

这意味着它只是遵从了CPU版本的LAPACK和GPU版本的CUBLAS:我认为这就是包装器代码:@Multihunter我自己的调查导致了代码中的同一点