jax.lax.linalg.hessenberg

jax.lax.linalg.hessenberg(a)

将方阵简化为上海森伯格形式。

目前仅在 CPU 上实现。

参数:

a (ArrayLike) – 浮点或复数方阵或矩阵批次。

返回:

一个 (a, taus) 对，其中 a 的上三角和第一条子对角线包含上海森伯格矩阵，第一条子对角线下方的元素包含 Householder 反射器。对于每个 Householder 反射器，taus 包含基本 Householder 反射器的标量因子。

返回类型:

tuple[Array, Array]