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 (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。