jax.lax.linalg.hessenberg

jax.lax.linalg.hessenberg(a)

将一个方阵简化为上Hessenberg形式。

目前仅在CPU上实现。

参数:

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

返回类型:

tuple[Array, Array]

返回值：一个 (a, taus) 对，其中 a 的上三角部分和第一子对角线包含上 Hessenberg 矩阵，而第一子对角线下方的元素包含 Householder 反射器。对于每个 Householder 反射器，taus 包含基本 Householder 反射器的标量因子。