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 反射器的純量因子。