c++ - 如何将 Eigen FFT 与 MatrixXf 一起使用？

Question

我是 Eigen 库的新手。我想计算特征矩阵的 FFT。但是，我这样做的尝试表明不受支持的 Eigen FFT 模块不能与 MatrixXf 一起使用。我想完成类似的事情:

#include <eigen3/unsupported/Eigen/FFT>
#include<Eigen/Dense>
#include<iostream>
using namespace std;
using namespace Eigen;
int main(){
    MatrixXf A = MatrixXf::Random(3,10);
    FFT<float> fft;
    MatrixXf B;
    fft.fwd(B,A);
}

这可以实现吗？欢迎任何其他建议。从 matlab 迁移到 Eigen 花费了我大量的 self 说服力，除非不可避免，否则我宁愿不使用其他库。谢谢。

Answer 1

不幸的是，它是不正确的；

1) 你必须迭代输入矩阵的行(真实的)

2) 然后遍历输出矩阵的列(复杂)

FFT<float> fft;
Eigen::Matrix<float, dim_x, dim_y> in = setMatrix();
Eigen::Matrix<complex<float>, dim_x, dim_y> out;

for (int k = 0; k < in.rows(); k++) {
    Eigen::Matrix<complex<float>, dim_x, 1> tmpOut;
    fft.fwd(tmpOut, in.row(k));
    out.row(k) = tmpOut;
}

for (int k = 0; k < in.cols(); k++) {
    Eigen::Matrix<complex<float>, 1, dim_y> tmpOut;
    fft.fwd(tmpOut, out.col(k));
    out.col(k) = tmpOut;
}