def jonker_compact_batch_unpacked(cost: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor]
Source: torchmatch/assignment/ops.py:11