tfp.experimental.fastgp.partial_lanczos.my_tridiagonal_solve

Like jax.linalg.tridiagonal_solve, but works for all sizes.

View aliases

Main aliases

`tfp.experimental.fastgp.fast_gp.fast_log_det.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.fast_gp.fast_log_det.preconditioners.linalg.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.fast_gp.preconditioners.linalg.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.fast_log_det.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.fast_log_det.preconditioners.linalg.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.fast_mtgp.fast_gp.fast_log_det.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.fast_mtgp.fast_gp.fast_log_det.preconditioners.linalg.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.fast_mtgp.fast_gp.preconditioners.linalg.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.fast_mtgp.fast_log_det.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.fast_mtgp.fast_log_det.preconditioners.linalg.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.fast_mtgp.preconditioners.linalg.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.linalg.partial_lanczos.my_tridiagonal_solve`, `tfp.experimental.fastgp.preconditioners.linalg.partial_lanczos.my_tridiagonal_solve`

tfp.experimental.fastgp.partial_lanczos.my_tridiagonal_solve(
    lower_diagonal: tfp.experimental.fastgp.fast_gp.Array,
    middle_diagonal: tfp.experimental.fastgp.fast_gp.Array,
    upper_diagonal: tfp.experimental.fastgp.fast_gp.Array,
    b: tfp.experimental.fastgp.fast_gp.Array
) -> tfp.experimental.fastgp.fast_gp.Array