jax.lax.linalg.cholesky_update

jax.lax.linalg.cholesky_update(r_matrix, w_vector)

Cholesky 秩-1 更新。

给定 Cholesky 分解 \(A = R.T \, R\) 和一个向量 \(w\)，计算 \(A + w \, w.T\) 的 Cholesky 分解，时间复杂度为 \(O(N^2)\)。

参数:

• r_matrix (ArrayLike) – 一个上三角矩阵 (R)，使得 \(A = R^T \, R\)。

• w_vector (ArrayLike) – 用于秩-1 更新的向量 \(w\)。

返回:

一个新的上三角矩阵 \(R\)，定义 \(A + w \, w^T\) 的 Cholesky 分解。

返回类型:

Array