Method trace

ot/backend.py:1403–1404

(self, a)

        return np.linalg.solve(a, b)

    def trace(self, a):
        return np.einsum("...ii", a)

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