Method trace

ot/backend.py:2814–2815

(self, a)

        return cp.linalg.solve(a, b)

    def trace(self, a):
        return cp.trace(a, axis1=-2, axis2=-1)

    def inv(self, a):
        return cp.linalg.inv(a)