C中的矩阵乘法错误

Question

我一直在编写一些代码来使用 C 实现离散余弦变换。这是我用于矩阵乘法的代码。

float Yt[8][8][4];
float Y8[8][8][4]={{{0.980785,-0.555570,0.555570 ...}}};
float A[8][8]={{0.707107,0.707107 ... }};

for(k=0;k<4;++k){
        for(i=0;i<8;++i){
            for(j=0;j<8;++j){
            Yt[i][j][k]=0;
            }}}

for(k=0;k<4;++k){
    for(i=0;i<8;++i){
         for(j=0;j<8;++j){
            for(l=0;l<8;++l){
                Yt[i][j][k]+=A[i][l]*Y8[l][j][k];
                }}}}

Y8 是同一个 8x8 矩阵，重复 4 次。 我的代码成功编译并运行。但是，我收到的输出显然是不正确的。 Yt[i][j][k] for k=0,1,2,3 应该都是相同的值，但我最终得到 k=0,1 的正确值和 k=2,3 的不同值。

谁能明白为什么我得到的最后两个矩阵的值不正确？

Answer 1

@chux 在使用之前对它们的值进行了很好的初始化Yt。你可以这样做：

for(k=0;k<4;++k){
  for(i=0;i<8;++i){
    for(j=0;j<8;++j){
      double temp = 0.0;
      for(l=0;l<8;++l){
        temp += A[i][l]*Y8[l][j][k];
      }
      Yt[i][j][k] = temp;
    }
  }
}

这将允许编译器使用寄存器进行累加，然后只访问一次Yt（及其三个索引操作）。

Answer 2

初始化Yt[i][j][k]。

我怀疑这将按如下方式完成

for(j=0;j<8;++j){
  Yt[i][j][k] = 0.0;
  for(l=0;l<8;++l){
    Yt[i][j][k] += A[i][l]*Y8[l][j][k];
  }
}

... 或在其声明中。类似double Yt[8][8][4] = { 0.0 };