dc39188e6611f254dfb9ef0358570efe59704ff5,geomstats/spd_matrices_space.py,SPDMatricesSpace,sqrtm,#SPDMatricesSpace#Any#,152

Before Change

        //                      vectors, diag, vectors)

        sqrt_mat = np.zeros_like(sym_mat)
        for i in range(sqrt_mat.shape[0]):
            sqrt_mat[i] = scipy.linalg.sqrtm(sym_mat[i])

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

    def random_uniform(self, n_samples=1):

After Change

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

        sqrt_eigenvalues = np.sqrt(eigenvalues)
        diag = np.diag(np.squeeze(sqrt_eigenvalues, axis=0))
        diag = vectorization.to_ndarray(diag, to_ndim=3)

        sqrt_mat = np.einsum("ijk,ikl,iml->ijm",
                             vectors, diag, vectors)

Instances