numpy.linalg.svd

linalg.svd(a, full_matrices=True, compute_uv=True, hermitian=False)

奇异值分解。

当 a 是一个二维数组，并且 full_matrices=False 时，它被分解为 u @ np.diag(s) @ vh = (u * s) @ vh，其中 u 和 vh 的共轭转置是具有正交列的二维数组，而 s 是 a 的奇异值的 1D 数组。当 a 是更高维时，SVD 将以堆叠模式应用，如下所述。

参数:

a(…, M, N) array_like

实数或复数数组，其中 a.ndim >= 2。

full_matricesbool, 可选

如果为 True（默认），则 u 和 vh 的形状分别为 (..., M, M) 和 (..., N, N)。否则，形状分别为 (..., M, K) 和 (..., K, N)，其中 K = min(M, N)。

compute_uvbool, 可选

是否除了 s 之外还计算 u 和 vh。默认值为 True。

hermitianbool, 可选

如果为 True，则假定 a 为 Hermitian（如果为实数，则为对称），可以启用更有效的奇异值查找方法。默认为 False。

版本 1.17.0 中的新内容。

返回值:

当 compute_uv 为 True 时，结果是一个命名元组，具有以下

属性名称

U{ (…, M, M), (…, M, K) } array

酉数组。前 a.ndim - 2 个维度与输入 a 的维度大小相同。最后两个维度的尺寸取决于 full_matrices 的值。仅当 compute_uv 为 True 时才返回。

S(…, K) array

包含奇异值的向量，在每个向量内按降序排列。前 a.ndim - 2 个维度与输入 a 的维度大小相同。

Vh{ (…, N, N), (…, K, N) } array

酉数组。前 a.ndim - 2 个维度与输入 a 的维度大小相同。最后两个维度的尺寸取决于 full_matrices 的值。仅当 compute_uv 为 True 时才返回。

引发:

LinAlgError

如果 SVD 计算不收敛。

另请参阅

scipy.linalg.svd

SciPy 中的类似函数。

scipy.linalg.svdvals

计算矩阵的奇异值。

注释

版本 1.8.0 中的更改: 广播规则适用，有关详细信息，请参阅 numpy.linalg 文档。

分解使用 LAPACK 例程 _gesdd 执行。

SVD 通常用于二维矩阵 \(A\) 的分解。更高维的情况将在下面讨论。在二维情况下，SVD 写成 \(A = U S V^H\)，其中 \(A = a\)，\(U= u\)，\(S= \mathtt{np.diag}(s)\) 且 \(V^H = vh\)。1D 数组 s 包含 a 的奇异值，而 u 和 vh 是酉的。 vh 的行是 \(A^H A\) 的特征向量，而 u 的列是 \(A A^H\) 的特征向量。在这两种情况下，相应的（可能非零）特征值由 s**2 给出。

如果 a 的维度超过两个，则广播规则适用，如 对多个矩阵同时进行线性代数运算 中所述。这意味着 SVD 正在“堆叠”模式下工作：它遍历前 a.ndim - 2 个维度的所有索引，并针对每个组合，SVD 将应用于最后两个索引。矩阵 a 可以使用 (u * s[..., None, :]) @ vh 或 u @ (s[..., None] * vh) 从分解中重建。（对于低于 3.5 的 python 版本，@ 运算符可以被函数 np.matmul 替换。）

如果 a 是一个 matrix 对象（而不是 ndarray），则所有返回值也是如此。

示例

>>> import numpy as np
>>> rng = np.random.default_rng()
>>> a = rng.normal(size=(9, 6)) + 1j*rng.normal(size=(9, 6))
>>> b = rng.normal(size=(2, 7, 8, 3)) + 1j*rng.normal(size=(2, 7, 8, 3))

基于完整 SVD 的重建，二维情况

>>> U, S, Vh = np.linalg.svd(a, full_matrices=True)
>>> U.shape, S.shape, Vh.shape
((9, 9), (6,), (6, 6))
>>> np.allclose(a, np.dot(U[:, :6] * S, Vh))
True
>>> smat = np.zeros((9, 6), dtype=complex)
>>> smat[:6, :6] = np.diag(S)
>>> np.allclose(a, np.dot(U, np.dot(smat, Vh)))
True

基于缩减 SVD 的重建，二维情况

>>> U, S, Vh = np.linalg.svd(a, full_matrices=False)
>>> U.shape, S.shape, Vh.shape
((9, 6), (6,), (6, 6))
>>> np.allclose(a, np.dot(U * S, Vh))
True
>>> smat = np.diag(S)
>>> np.allclose(a, np.dot(U, np.dot(smat, Vh)))
True

基于完整 SVD 的重建，四维情况

>>> U, S, Vh = np.linalg.svd(b, full_matrices=True)
>>> U.shape, S.shape, Vh.shape
((2, 7, 8, 8), (2, 7, 3), (2, 7, 3, 3))
>>> np.allclose(b, np.matmul(U[..., :3] * S[..., None, :], Vh))
True
>>> np.allclose(b, np.matmul(U[..., :3], S[..., None] * Vh))
True

基于缩减 SVD 的重建，四维情况

>>> U, S, Vh = np.linalg.svd(b, full_matrices=False)
>>> U.shape, S.shape, Vh.shape
((2, 7, 8, 3), (2, 7, 3), (2, 7, 3, 3))
>>> np.allclose(b, np.matmul(U * S[..., None, :], Vh))
True
>>> np.allclose(b, np.matmul(U, S[..., None] * Vh))
True