jax.lax.linalg.householder_product

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

基本豪斯霍爾德反射器的乘積。

參數:

• a (ArrayLike) – 形狀為 [..., m, n] 的矩陣，其下三角包含基本豪斯霍爾德反射器。

• taus (ArrayLike) – 形狀為 [..., k] 的向量，其中 k < min(m, n)，包含基本豪斯霍爾德反射器的純量因子。

返回:

一批正交（么正）矩陣，其形狀與 a 相同，包含基本豪斯霍爾德反射器的乘積。

返回類型:

Array