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 是一个酉矩阵（正交矩阵），如果 full_matrices=True，则形状为 [..., m, m]；如果 full_matrices=False，则形状为 [..., m, min(m, n)]。

数组 r 是一个上三角矩阵，如果 full_matrices=True，则形状为 [..., m, n]；如果 full_matrices=False，则形状为 [..., min(m, n), n]。

返回类型:

tuple[Array, Array]