tfp.experimental.fastgp.partial_lanczos.my_tridiagonal_solve

View source on GitHub

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