jax.lax.linalg.qdwh

jax.lax.linalg.qdwh(x, *, is_hermitian=False, max_iterations=None, eps=None, dynamic_shape=None)

基于QR的动态加权Halley迭代用于极分解。

参数:

• x – 一个满秩矩阵，形状为 M x N。该矩阵可能从较小的真实形状（dynamic_shape）填充到该大小。

• is_hermitian (bool) – 如果 x 是厄米特矩阵则为真。默认为 False。此参数当前未使用，但存在以保持向后兼容性。

• eps (float | None) – 最终结果将满足 |x_k - x_k-1| < |x_k| * (4*eps)**(1/3) 其中 x_k 是迭代值。

• max_iterations (int | None) – 即使上述条件未满足，迭代也将在这么多步骤后终止。

• dynamic_shape (tuple[int, int] | None) – 未填充形状作为 (m, n) 元组；可选。

返回:

一个四元组 (u, h, num_iters, is_converged)，包含 x = u * h 的极分解、计算 u 所需的迭代次数，以及 is_converged，当在最大迭代次数内达到收敛时，其值为 True。