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) 元組；選用。

傳回值:

包含 x = u * h 的極分解的四元組 (u, h, num_iters, is_converged)、計算 u 的迭代次數，以及 is_converged，當在最大迭代次數內達到收斂時，其值為 True。