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]
。- 返回类型: