transport.samples.kernels.cg_dense.materialize_transport_plan
def materialize_transport_plan(x: torch.Tensor, y: torch.Tensor, f_hat: torch.Tensor, g_hat: torch.Tensor, eps: float, x2: torch.Tensor | None = None, y2: torch.Tensor | None = None) -> torch.TensorMaterialize the transport plan P = exp((f_hat + g_hat - C) / eps).
Parameters
| Name | Type | Description |
|---|---|---|
| x | torch.Tensor | Source points (n, d) |
| y | torch.Tensor | Target points (m, d) |
| f_hat | torch.Tensor | Source raw-form potential (n,) |
| g_hat | torch.Tensor | Target raw-form potential (m,) |
| eps | float | Entropy regularization |
| x2 = None | torch.Tensor | None | Precomputed ||x_i||² (n,), optional |
| y2 = None | torch.Tensor | None | Precomputed ||y_j||² (m,), optional |
Returns
torch.Tensor — Transport plan matrix (n, m), dtype=float32
Source: torchmatch/transport/samples/kernels/cg_dense.py:39