mindspore.ops.svd

mindspore.ops.svd(a, full_matrices=False, compute_uv=True)[源代码]

计算单个或多个矩阵的奇异值分解。

设矩阵 \(A\) ，svd返回奇异值 \(S\) 、左奇异向量 \(U\) 和右奇异向量 \(V\) 。满足以下公式:

\[A=U*diag(S)*V^{T}\]
参数：

  • a (Tensor) - 待分解的矩阵。shape为 \((*, M, N)\) ，支持的数据类型为float32和float64。

  • full_matrices (bool, 可选) - 如果为True，则计算完整的 \(U\) 和 \(V\) 。否则仅计算前P个奇异向量，P为M和N中的较小值，M和N分别是输入矩阵的行和列。默认值：False。

  • compute_uv (bool, 可选) - 如果这个参数为True，则计算 \(U\) 和 \(V\) ，否则只计算 \(S\) 。默认值：True。

返回：

  • s (Tensor) - 奇异值。shape为 \((*, P)\) 。

  • u (Tensor) - 左奇异向量。如果 compute_uv 为False，该值不会返回。shape为 \((*, M, P)\) 。如果 full_matrices 为True，则shape为 \((*, M, M)\) 。

  • v (Tensor) - 右奇异向量。如果 compute_uv 为False，该值不会返回。shape为 \((*, N, P)\) 。如果 full_matrices 为True，则shape为 \((*, N, N)\) 。

异常：

  • TypeError - full_matrices 或 compute_uv 不是bool类型。

  • TypeError - 输入的rank小于2。

  • TypeError - 输入的数据类型不为float32或float64。

支持平台：

GPU CPU

样例：

>>> import numpy as np
>>> from mindspore import Tensor, set_context
>>> from mindspore import ops
>>> set_context(device_target="CPU")
>>> a = Tensor(np.array([[1, 2], [-4, -5], [2, 1]]).astype(np.float32))
>>> s, u, v = ops.svd(a, full_matrices=True, compute_uv=True)
>>> print(s)
[7.0652843 1.040081 ]
>>> print(u)
[[ 0.30821905 -0.48819482 0.81649697]
 [-0.90613353  0.11070572 0.40824813]
 [ 0.2896955   0.8656849  0.4082479 ]]
>>> print(v)
[[ 0.63863593 0.769509  ]
 [ 0.769509  -0.63863593]]