jax.lax.linalg.householder_product

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

初等 Householder 反射子的乘积。

参数:

• a (ArrayLike) – 形状为 [..., m, n] 的矩阵，其下三角包含初等 Householder 反射子。

• taus (ArrayLike) – 形状为 [..., k] 的向量，其中 k < min(m, n)，包含初等 Householder 反射子的标量因子。

返回:

与 a 具有相同形状的一批正交（酉）矩阵，包含初等 Householder 反射子的乘积。

返回类型:

Array