jax.lax.linalg.tridiagonal

jax.lax.linalg.tridiagonal(a, *, lower=True)

将一个对称/厄米矩阵简化为三对角形式。

目前仅在CPU和GPU上实现。

参数:

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

• lower – 描述使用输入矩阵的哪个三角形。其他三角形被忽略且不会被访问。

返回类型:

tuple[Array, Array, Array, Array]

返回: 一个 (a, d, e, taus) 对。如果 lower=True，矩阵（或矩阵批次）``a`` 的对角线和第一子对角线包含三对角表示，第一子对角线以下的元素包含基本 Householder 反射器，其中 d 包含矩阵的对角线，e 包含第一子对角线。如果 lower=False，矩阵的对角线和第一超对角线包含三对角表示，第一超对角线以上的元素包含基本 Householder 反射器，其中 d 包含矩阵的对角线，e 包含第一超对角线。taus 包含基本 Householder 反射器的标量因子。