View source on GitHub
|
Approximate log det using a 6th order rational function.
tfp.experimental.fastgp.fast_log_det.r6(
M: tf2jax.linalg.LinearOperator,
preconditioner: tfp.experimental.fastgp.preconditioners.Preconditioner,
key: tfp.experimental.fastgp.fast_gp.Array,
num_probe_vectors: int = 25,
probe_vector_type: tfp.experimental.fastgp.ProbeVectorType = ProbeVectorType.RADEMACHER,
num_iters: int = 20,
**unused_kwargs
)
View source on GitHub