transport.matrix.ops.exact_emd

def exact_emd(cost: torch.Tensor, mask: torch.Tensor | None = None, a: torch.Tensor | None = None, b: torch.Tensor | None = None) -> torch.Tensor