jax.lax.linalg.qr

jax.lax.linalg.qr(x, *, full_matrices=True)

QR 分解。

计算QR分解

\[A = Q . R\]

矩阵 \(A\) 的分解，使得 \(Q\) 是一个酉（正交）矩阵，而 \(R\) 是一个上三角矩阵。

参数:

• x (ArrayLike) – 一批形状为 [..., m, n] 的矩阵。

• full_matrices (bool) – 确定返回完整矩阵还是简化矩阵；见下文。

返回:

一对数组 (q, r)。数组 q 是一个酉（正交）矩阵，形状为 [..., m, m] 如果 full_matrices=True，或者 [..., m, min(m, n)] 如果 full_matrices=False。数组 r 是一个上三角矩阵，形状为 [..., m, n] 如果 full_matrices=True，或者 [..., min(m, n), n] 如果 full_matrices=False。

返回类型:

tuple[Array, Array]