jax.lax.linalg.qdwh

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

基于 QR 的极分解动态加权哈雷迭代。

参数:

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