Question

我有一个真正的2D矩阵。我正在使用fftw来获取它的fft。但是使用真实到复合fft的结果不同于复数（假想部分等于零）到复数fft。

真实矩阵

 0     1     2
 3     4     5
 6     7     8

真实到复杂fft的结果

36 -4.5+2.59808i  -13.5+7.79423i 
0  -13.5-7.79423i 0 
0  0              0

代码：

int r = 3, c = 3;
int sz = r * c;
double *in = (double*) malloc(sizeof(double) * sz);
fftw_complex *out = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * sz);
fftw_plan p = fftw_plan_dft_r2c_2d(r, c, in, out, FFTW_MEASURE);
for ( int i=0; i<r; ++i ){
    for ( int j=0; j<c; ++j ){
        in[i*c+j] = i*c + j;
    }
}
fftw_execute(p);

使用虚部为零的复杂矩阵

复杂矩阵

 0+0i     1+0i     2+0i
 3+0i     4+0i     5+0i
 6+0i     7+0i     8+0i

复杂到复杂fft的结果

36               -4.5 + 2.59808i  -4.5 - 2.59808i 
-13.5 + 7.79423i 0               0 
-13.5 - 7.79423i 0               0

代码：

int r = 3, c = 3;
int sz = r * c;
fftw_complex *out = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * sz);
fftw_complex *inc = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * sz);
p = fftw_plan_dft_2d( r,c, inc, out, FFTW_FORWARD,FFTW_MEASURE);
for ( int i=0; i<r; ++i ){
    for ( int j=0; j<c; ++j ){
        inc[i*c+j][0] = i*c+j;
        inc[i*c+j][1] = 0;
    }
}
fftw_execute(p);

我是复杂到复杂fft的结果。但真正的复杂fft更快，我的数据是真实的。我是编程错误还是结果不同？

Answer 1

如FFTW documentation

所示

然后，在r2c变换之后，输出为n ₀×n ₁×n ₂×...×（n _d-1 / 2 + 1）行主要顺序中的fftw_complex值数组

换句话说，样本实矩阵的实数到复数变换的输出实际上是：

 36            -4.5+2.59808i
-13.5+7.79423i  0
-13.5-7.79423i  0

您可能会注意到这两列完全匹配复杂到复杂变换的前两列。从实对复变换中省略了缺失列，因为由于对称性它是冗余的。因此，可以使用以下方法构建包括缺失列的完整3x3矩阵：

fftw_complex *outfull = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * sz);
int outc = (c/2+1);
for ( int i=0; i<r; ++i ){
    // copy existing columns
    for ( int j=0; j<outc; ++j ){
        outfull[i*c+j][0] = out[i*outc+j][0];
        outfull[i*c+j][1] = out[i*outc+j][1];
    }
    // generate missing column(s) from symmetry
    for ( int j=outc; j<c; ++j){
        int row = (r-i)%r;
        int col = c-j;
        outfull[i*c+j][0] =  out[row*outc+col][0];
        outfull[i*c+j][1] = -out[row*outc+col][1];
    }
}

真实fft和复数fft之间的差异fftw中虚部为零？

1 个答案: