jax.lax.linalg.householder_乘积

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

基本Householder反射器的乘积。

参数:

• a (ArrayLike) – 一个形状为 [..., m, n] 的矩阵，其下三角包含基本的 Householder 反射器。

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

返回:

一批与 a 形状相同的正交（酉）矩阵，包含基本Householder反射器的乘积。

返回类型:

Array