ott.tools.unreg.hungarian

Contents

ott.tools.unreg.hungarian#

ott.tools.unreg.hungarian(geom)[source]#

Solve matching problem using Hungarian algorithm from optax.

Parameters:: geom (Geometry) – Geometry object with square (shape [n,n]) cost matrix.
Return type:: Tuple[Array, HungarianOutput]
Returns:: The value of the unregularized OT problem, along with an output object listing relevant information on outputs.