transport.matrix.solve

def solve(cost: torch.Tensor, backend: Backend | str = Backend.AUTO, reg: float = 0.1, n_iter: int = 100, mask: torch.Tensor | None = None, a: torch.Tensor | None = None, b: torch.Tensor | None = None, scaling: float | None = None, rho: float = 1.0, cost_aa: torch.Tensor | None = None, cost_bb: torch.Tensor | None = None, unpack: bool = False) -> torch.Tensor | tuple[torch.Tensor, torch.Tensor | None, torch.Tensor | None]