function

assignment.lap.lap_batch_assignment

def lap_batch_assignment(cost_matrices: list[torch.Tensor]) -> list[tuple[torch.Tensor, torch.Tensor, torch.Tensor]]

Solve a batch of linear assignment problems via torchmatch CUDA solvers.

Parameters

NameTypeDescription
cost_matriceslist[torch.Tensor]List of ``(N_i, M_i)`` cost matrices.

Returns

list of tuple — Per-problem ``(matches, unmatched_rows, unmatched_cols)`` triples following the shape conventions of :func:`lap_assignment`.

Source: unitrack/assignment/lap/_solver.py:158