pygmtools.pytorch_backend.sinkhorn

pygmtools.pytorch_backend.sinkhorn(s: torch.Tensor, nrows: Optional[torch.Tensor] = None, ncols: Optional[torch.Tensor] = None, dummy_row: bool = False, max_iter: int = 10, tau: float = 1.0, batched_operation: bool = False) → torch.Tensor

Pytorch implementation of Sinkhorn algorithm