torchmatch
property

transport.samples.kernels.cg_dense.CachedDenseHVP.a_hat

a_hat = torch.mv(self.P, ones_m)

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