如何解决如何找到torch.solve的源代码?
我担心torch.solve()
是否检查线性系统的系数矩阵的条件并采用理想的预处理条件;因此,我对其实现细节感到好奇。我已经阅读了几个试图追踪源文件但徒劳的答案。我希望有人能帮助我在ATen库中找到它的定义。
解决方法
我认为它只是将LAPACK用于CPU,将CUBLAS用于GPU,因为torch.solve
在official docs的“ BLAS和LAPACK操作”下列出。
然后,我们正在寻找包装程序代码,我相信它是this part。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。