torchmatch
property

transport.samples.kernels.cg_dense.CachedDenseHVP.diag_y

diag_y = diag_factor_y * b_hat_clamped

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