assignment.Assignment.forward
def forward(self, cost_matrix: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor, torch.Tensor]Solve the cost matrix.
Parameters
| Name | Type | Description |
|---|---|---|
| cost_matrix | torch.Tensor | ``(N, M)`` cost matrix to solve. |
Returns
torch.Tensor — ``(K, 2)`` long tensor. Column 0 holds tracklet (row) indices, column 1 holds detection (column) indices.
Source: unitrack/assignment/_base.py:35