Question

我正在使用犰狳。我有这些变量：

arma::Mat<float> m_matrix;
arma::cc_vec     m_eigenvalues;
arma::cx_mat     m_eigenvectors;

我想这样做：

void calculate_eigens ()
{
    arma :: eig_gen (m_eigenvalues, m_eigenvectors, m_matrix);
}

但是eig_gen的函数原型不允许这些参数类型，它希望第三个参数是双精度矩阵。

以下是gcc错误：

错误：没有匹配函数来调用'eig_gen（arma :: cx_vec＆amp;，arma :: cx_mat＆amp;，arma :: Mat＆amp;）'

注意：候选人：模板类型名称arma :: enable_if2 :: value，arma :: Col＆gt; ＆gt; ::结果arma :: eig_gen（const arma :: Base＆amp;）

armadillo_bits / fn_eig_gen.hpp:25：1：注意：模板参数扣除/替换失败：

注意：候选人需要1个参数，3个提供

这解决了它。

void calculate_eigens ()
{
    auto tmp = arma::conv_to<arma::Mat<double>>::from(m_matrix);

    arma :: eig_gen (m_eigenvalues, m_eigenvectors, tmp);
}

我不想做这个转换。我查看了犰狳的来源，似乎eig_gen将其主要工作推迟到这个功能：

  template<typename eT>
  inline
  void
  geev(char* jobvl, char* jobvr, blas_int* N, eT* a, blas_int* lda, eT* wr, eT* wi, eT* vl, blas_int* ldvl, eT* vr, blas_int* ldvr, eT* work, blas_int* lwork, blas_int* info)
    {
    arma_type_check(( is_supported_blas_type<eT>::value == false ));

    if(is_float<eT>::value)
      {
      typedef float T;
      arma_fortran(arma_sgeev)(jobvl, jobvr, N, (T*)a, lda, (T*)wr, (T*)wi, (T*)vl, ldvl, (T*)vr, ldvr, (T*)work, lwork, info);
      }
    else
    if(is_double<eT>::value)
      {
      typedef double T;
      arma_fortran(arma_dgeev)(jobvl, jobvr, N, (T*)a, lda, (T*)wr, (T*)wi, (T*)vl, ldvl, (T*)vr, ldvr, (T*)work, lwork, info);
      }
    }

eT*a参数是从Mat<T>::get_ref m_matrix传来的，所以在我看来，图书馆作者希望eig_gen能够在Mat<float>上愉快地工作}以及Mat<double>但它在实践中不起作用。

如何在Mat<float>上计算特征值/特征向量而不先转换为Mat<double>？

Answer 1

您需要确保所有矩阵和向量具有相同的精度，而不仅仅是输入矩阵。单精度的正确用法是：

arma::fmat     X;
arma::cx_fvec  eigvals;
arma::cx_fmat  eigvecs;

// ... fill X with values ...

arma::eig_gen(eigvals, eigvecs, X);

犰狳可以在Mat <float>上做eig_gen吗？

1 个答案: