assignment
module torchmatch.assignmentInteger linear assignment problem (LAP) solvers.
Public surface:
- :func:`solve` -- the dispatcher; resolves AUTO at call time so the
picked op traces cleanly under ``torch.compile``.
- :class:`Backend` -- backend choices accepted by :func:`solve`.
- :mod:`ops` -- direct access to ``torch.ops.assignment.*`` op handles
for callers that want to pin a backend.
Members
module
- opsDirect access to ``torch.ops.assignment.*`` ops.
type
function
- assignment_costCompute the total cost of a LAP assignment.
- auction_assignmentSolve a linear assignment problem using Bertsekas' auction algorithm.
- load_cpuRegister ``torch.ops.assignment.jonker_*`` (CPU backend).
- load_cudaRegister the CUDA-only ops in ``torch.ops.assignment``.
- resolve_backendReturn the op name that ``solve()`` would dispatch to.
- solve
Source: torchmatch/assignment/__init__.py:1