犰狳复矩阵上第 2 个暗淡的 1D fft

Question

我需要计算 cx_mat 在它的第二维中的一维 fft。所以说我有这个矩阵：

cx_mat A(randu(5,10),randu(5,10));

MATLAB 版本如下所示：

A_fft=fft(A,[],2);

我将如何在犰狳中解决这个问题？

我试图尽可能快地保持这一点，因此我认为在列中运行的 for 循环不是最佳选择。

然后我看到了.each_col 属性并尝试了

cx_mat A_fft=A.each_col([](vec& a){fft(a);});

但这不会编译。也许这是正确的，但我的语法是错误的

任何帮助将不胜感激。

Answer 1

您需要有一个复数向量作为 lambda 函数中的参数： cx_mat A_fft=A.each_col([](cx_vec& a){fft(a);});