jax.lax.linalg.qr

内容

jax.lax.linalg.qr#

jax.lax.linalg.qr(x, *, full_matrices=True)[source]#

QR 分解。

计算 QR 分解

\[A = Q . R\]

矩阵 \(A\),使得 \(Q\) 是一个酉(正交)矩阵,而 \(R\) 是一个上三角矩阵。

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

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

返回:

一对数组 (q, r)

如果 full_matrices=True,则数组 q 是一个酉(正交)矩阵,形状为 [..., m, m],如果 full_matrices=False,则形状为 [..., m, min(m, n)]

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

返回类型:

元组[数组, 数组]