assignment.Assignment
class Assignment(torch.nn.Module):Base class for linear-assignment-problem solvers.
Subclasses implement ``_assign`` over a 2-D cost matrix. The base
:meth:`forward` masks entries above :attr:`threshold` to ``inf``
before dispatching, so callers can bound match cost without each
backend re-implementing the guard.
Parameters
| Name | Type | Description |
|---|---|---|
| threshold = torch.inf | float | Cost upper bound. Entries strictly above ``threshold`` are replaced with ``inf`` and treated as forbidden by the solver. Default ``inf`` (no thresholding). |
Members
property
Source: unitrack/assignment/_base.py:10