jax.lax.linalg.householder_product

jax.lax.linalg.householder_product(a, taus)

Product of elementary Householder reflectors.

Parameters:

• a (ArrayLike) – A matrix with shape [..., m, n], whose lower triangle contains elementary Householder reflectors.

• taus (ArrayLike) – A vector with shape [..., k], where k < min(m, n), containing the scalar factors of the elementary Householder reflectors.

Returns:

A batch of orthogonal (unitary) matrices with the same shape as a, containing the products of the elementary Householder reflectors.

Return type:

Array