31d8076c8dd31c28054e820571ef38234950e101,geomstats/spd_matrices_space.py,,group_log,#Any#,76

Before Change

    diag_log = np.zeros((n_sym_mats, mat_dim, mat_dim))
    for i in range(n_sym_mats):
        diag_log[i] = np.diag(np.log(eigenvalues[i]))

    log = np.matmul(diag_log, np.transpose(vectors, axes=(0, 2, 1)))
    log = np.matmul(vectors, log)
    return log

After Change

    log_eigenvalues = np.log(eigenvalues)

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

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


class SPDMatricesSpace(EmbeddedManifold):

Instances