Method solve

ot/backend.py:2385–2386

(self, a, b)

        return results

    def solve(self, a, b):
        return torch.linalg.solve(a, b)

    def trace(self, a):
        return torch.diagonal(a, dim1=-2, dim2=-1).sum(-1)