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 是埃尔米特矩阵，则为 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。