fa1baf601959e2add7f333540247f85ba5d921fc,geomstats/_backend/numpy/linalg.py,,logm,#Any#,71

Before Change

        eigvals, eigvecs = np.linalg.eigh(new_x)
        if (eigvals > 0).all():
            eigvals = np.log(eigvals)
            eigvals = np.vectorize(np.diag, signature="(n)->(n,n)")(eigvals)
            transp_eigvecs = np.transpose(eigvecs, axes=(0, 2, 1))
            result = np.matmul(eigvecs, eigvals)
            result = np.matmul(result, transp_eigvecs)
        else:
            result = np.vectorize(scipy.linalg.logm,
                                  signature="(n,m)->(n,m)")(new_x)
    else:
        result = np.vectorize(scipy.linalg.logm,
                              signature="(n,m)->(n,m)")(new_x)

    if ndim == 2:
        return result[0]
    return result


def powerm(x, power):
    ndim = x.ndim

After Change

def logm(x):
    return np.vectorize(
        asp.logm, signature="(n,m)->(n,m)")(x)


def solve_sylvester(a, b, q):
    if a.shape == b.shape:

Instances