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 是 Hermitian 则为 True。默认为 False。此参数当前未使用，但为了向后兼容而存在。

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

• max_iterations (整数 | None) – 即使上述条件不满足，迭代也会在达到此步数后终止。

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

返回值:

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