ca7b264f69db393e3d1c704a20b4f22047d1998f,geomstats/spd_matrices_space.py,,group_exp,#Any#,41

Before Change

    [eigenvalues, vectors] = np.linalg.eigh(sym_mat)
    diag_exp = np.zeros((n_sym_mats, mat_dim, mat_dim))
    for i in range(n_sym_mats):
        diag_exp[i] = np.diag(np.exp(eigenvalues[i]))

    exp = np.matmul(diag_exp, np.transpose(vectors, axes=(0, 2, 1)))
    exp = np.matmul(vectors, exp)
    return exp

After Change

    [eigenvalues, vectors] = np.linalg.eigh(sym_mat)
    exp_eigenvalues = np.exp(eigenvalues)

    aux = np.einsum("ijk,ik->ijk", vectors, exp_eigenvalues)
    exp_mat = np.einsum("ijk,ilk->ijl", aux, vectors)

    exp_mat = vectorization.to_ndarray(exp_mat, to_ndim=3)
    return exp_mat


def group_log(sym_mat):

Instances