Ctrl+K

jax.lax.linalg.householder_product

jax.lax.linalg.householder_product#

jax.lax.linalg.householder_product(a, taus)[source]#

Product of elementary Householder reflectors.

Parameters:

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

  • taus (jax.typing.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

Contents