ott.tools.unreg.hungarian

ott.tools.unreg.hungarian(geom)

Solve matching problem using the Hungarian algorithm.

Uses the implementation from optax.

Parameters:

geom (Geometry) – Geometry object with square (shape [n, n]) cost_matrix.

Return type:

Tuple[Array, HardAssignmentOutput]

Returns:

The value of the unregularized OT problem, along with an output object listing relevant information on outputs.