SciPy SVD与Numpy SVD

A.M*_*.M. 9 python numpy scipy svd

SciPy和Numpy都内置了奇异值分解(SVD)函数.命令基本上scipy.linalg.svdnumpy.linalg.svd.这两者有什么区别?他们中的任何一个比另一个好吗?

Zic*_*ang 5

常见问题页面,它说scipy.linalg子模块为 Fortran LAPACK 库提供了一个更完整的包装器,同时numpy.linalg试图能够独立于 LAPACK 进行构建。

我对函数的不同实现做了一些基准测试svd,发现scipy.linalg.svd它比 numpy 对应的更快:

然而,jax包裹 numpy,又名jax.numpy.linalg.svd更快:

在此处获得用于基准测试的完整笔记本。