transport.samples.kernels.cg_dense.CachedDenseHVP.Py

Py = torch.mm(self.P, self.y_f)