jax.lax.linalg.hessenberg# jax.lax.linalg.hessenberg(a)[source]# 将方阵简化为上海森伯格形式。 目前仅在 CPU 上实现。 参数: a (ArrayLike) – 浮点数或复数方阵,或矩阵批次。 返回: 一个 (a, taus) 对,其中 a 的上三角和第一条次对角线包含上海森伯格矩阵,而第一条次对角线以下的元素包含 Householder 反射器。对于每个 Householder 反射器,taus 包含基本 Householder 反射器的标量因子。 返回类型: tuple[Array, Array]