View source on GitHub
|
Return a preconditioner as a linear operator.
tfp.experimental.fastgp.partial_lanczos.make_lanczos_preconditioner(
kernel: tf2jax.linalg.LinearOperator,
key: tfp.experimental.fastgp.fast_gp.Array,
num_iters: int = 20
)
View source on GitHub