View source on GitHub
|
Return low rank approximation to M based on the partial Lancozs alg.
tfp.experimental.fastgp.linalg.make_partial_lanczos(
key: tfp.experimental.fastgp.fast_gp.Array,
M: tf2jax.linalg.LinearOperator,
rank: int
) -> tfp.experimental.fastgp.fast_gp.Array
View source on GitHub