jax.lax.linalg.qdwh¶

jax.lax.linalg.qdwh(x, is_symmetric, max_iterations=10)[source]¶

QR-based dynamically weighted Halley iteration for polar decomposition.

Parameters

x – A full-rank matrix of shape m x n with m >= n.
is_symmetric – True if x is symmetric.
max_iterations – The predefined maximum number of iterations.

Returns

A four-tuple of (u, h, num_iters, is_converged) containing the polar decomposition of x = u * h, the number of iterations to compute u, and is_converged, whose value is True when the convergence is achieved within the maximum number of iterations.