jax.lax.linalg.tridiagonal#

jax.lax.linalg.tridiagonal(a, *, lower=True)[源代码]#

将对称/埃尔米特矩阵简化为三对角线形式。

目前仅在 CPU 和 GPU 上实现。

参数:

  • a (ArrayLike) – 浮点数或复数矩阵或批量矩阵。

  • lower (bool) – 描述要使用输入矩阵的哪个三角形。 另一个三角形将被忽略且不会被访问。

返回:

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

返回类型:

tuple[Array, Array, Array, Array]