transport.matrix.ops.log_sinkhorn

def log_sinkhorn(cost: torch.Tensor, eps: float, n_iter: int, a: torch.Tensor, b: torch.Tensor, mask: torch.Tensor | None = None, scaling: float | None = None) -> torch.Tensor