tfp.experimental.fastgp.preconditioners.PartialLanczosPlusScalingPreconditioner

https://www.sciencedirect.com/science/article/pii/S0307904X13002382 .

Inherits From: LowRankPlusScalingPreconditioner, Preconditioner

tfp.experimental.fastgp.preconditioners.PartialLanczosPlusScalingPreconditioner(
    M: tf2jax.linalg.LinearOperator,
    scaling: tfp.experimental.fastgp.fast_gp.Array,
    key: tfp.experimental.fastgp.fast_gp.Array,
    rank: int = 20,
    **unused_kwargs
)

Methods

from_lowrank

View source

@classmethod
from_lowrank(
    M, low_rank, scaling
)

Alternate constructor when low_rank is already made.

full_preconditioner

View source

full_preconditioner() -> tf2jax.linalg.LinearOperator

Returns the preconditioner.

log_det

View source

log_det() -> tf2jax.linalg.LinearOperator

The log absolute value of the determinant of the preconditioner.

preconditioned_operator

View source

preconditioned_operator() -> tf2jax.linalg.LinearOperator

Returns the combined action of M and the preconditioner.

trace_of_inverse_product

View source

trace_of_inverse_product(
    A: tfp.experimental.fastgp.fast_gp.Array
)

Returns tr( P^(-1) A ) for a n x n, non-batched A.

tree_flatten

View source

tree_flatten()

tree_unflatten

View source

@classmethod
tree_unflatten(
    unused_aux_data, children
)