cr.sparse.sls.ISTAState

class cr.sparse.sls.ISTAState(x: jax._src.numpy.lax_numpy.ndarray, r: jax._src.numpy.lax_numpy.ndarray, r_norm_sqr: jax._src.numpy.lax_numpy.ndarray, x_change_norm: jax._src.numpy.lax_numpy.ndarray, iterations: int)[source]

ISTA algorithm state

Attributes

iterations

Number of iterations to converge

r

Current residual

r_norm_sqr

Square of residual norm

x

Current solution estimate

x_change_norm

Change in the norm of x