assignment.lap.lap_assignment
def lap_assignment(cost_matrix: torch.Tensor, backend: Backend | str = Backend.CLASSICAL) -> tuple[torch.Tensor, torch.Tensor, torch.Tensor]Solve the linear assignment problem via a torchmatch CUDA solver.
Parameters
| Name | Type | Description |
|---|---|---|
| cost_matrix | torch.Tensor | ``(N, M)`` cost matrix. Non-finite entries (``inf``) mark forbidden assignments. |
| backend = Backend.CLASSICAL | Backend | str | Solver backend (default :attr:`Backend.CLASSICAL`). Accepts a :class:`Backend` member or its string value. |
Returns
torch.Tensor — ``(K, 2)`` long tensor of matched ``(row, col)`` indices.
Raises
- RuntimeError — If no CUDA device is available.
Source: unitrack/assignment/lap/_solver.py:89