torchmatch
property

transport.samples.kernels.cg_dense.CachedDenseHVP.b_hat

b_hat = torch.mv(self.P.t(), ones_n)

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