transport.samples.kernels.cg_dense.CachedDenseHVP.hvp
def hvp(self, A: torch.Tensor) -> tuple[torch.Tensor, DenseCgInfo]Compute HVP using cached transport plan.
Parameters
| Name | Type | Description |
|---|---|---|
| A | torch.Tensor | Input matrix for HVP (n, d) |
Returns
torch.Tensor — Hessian-vector product H @ A, shape (n, d)
Source: torchmatch/transport/samples/kernels/cg_dense.py:508