jax.lax.linalg.householder_product

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

Product of elementary Householder reflectors.

Parameters:

• a (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – A matrix with shape [..., m, n], whose lower triangle contains elementary Householder reflectors.

• taus (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – A vector with shape [..., k], where k < min(m, n), containing the scalar factors of the elementary Householder reflectors.

Return type:

Array

Returns:

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