77e56feb2a220db76da79613e346ffc5110061c0,geomstats/backend/numpy_linalg.py,,logm,#Any#,33

Before Change

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

After Change

    new_x = to_ndarray(x, to_ndim=3)
    if (new_x - np.transpose(new_x, axes=(0, 2, 1)) == 0).all():
        eigvals, eigvecs = np.linalg.eigh(new_x)
        eigvals = np.log(eigvals)
        if (eigvals > 0).all():
            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:

Instances