tf.contrib.kfac.utils.compute_pi

compute_pi(
    left_factor,
    right_factor
)

Defined in tensorflow/contrib/kfac/python/ops/utils.py.

Computes the scalar constant pi for Tikhonov regularization/damping.

pi = sqrt( (trace(A) / dim(A)) / (trace(B) / dim(B)) ) See section 6.3 of https://arxiv.org/pdf/1503.05671.pdf for details.

Args:

• left_factor: The left Kronecker factor Tensor.
• right_factor: The right Kronecker factor Tensor.

Returns:

The computed scalar constant pi for these Kronecker Factors (as a Tensor).