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 反射器的乘积。

返回类型：:

数组