506b234094009d55d326f1ce957f53f15d4b65ee,geomstats/spd_matrices_space.py,,group_log,#Any#,52

Before Change

    assert np.all(eigenvalues > 0)

    diag_log = np.diag(np.log(eigenvalues))
    log = np.dot(np.dot(vectors, diag_log), vectors.transpose())

    return log

After Change

    print(vectors.shape)
    assert np.all(eigenvalues > 0)
    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)
    print("log")
    print(log.shape)
    return log


class SPDMatricesSpace(Manifold):

Instances