torchmatch
function

assignment.ops.munkres

def munkres(cost: torch.Tensor) -> torch.Tensor

Source: torchmatch/assignment/ops.py:17