pygmtools.pytorch_backend.hungarian

pygmtools.pytorch_backend.hungarian(s: torch.Tensor, n1: Optional[torch.Tensor] = None, n2: Optional[torch.Tensor] = None, nproc: int = 1) → torch.Tensor

Pytorch implementation of Hungarian algorithm