torchmatch
property

transport.samples.kernels.cg_dense.CachedDenseHVP.diag_x

diag_x = diag_factor_x * a_hat_clamped

Source: torchmatch/transport/samples/kernels/cg_dense.py:496