C:矩阵 vector 乘积，两个双数相乘给出错误的符号

Question

我正在尝试执行一个简单的矩阵乘法 vector 乘法，但出于某种原因，我在几次乘法的结果中得到了错误的符号。我不知道为什么会这样，任何指针将不胜感激。

这是我的全部代码，即矩阵 * vector 函数和来自 mex 的调用函数(或任何调用函数)- 我正在通过 mex 从 Matlab 运行代码。

#include "mex.h"

void mxv(int m, int n, double *A, double *b, double *c) {
    double sum;
    int i, j;

    for (i = 0; i < m; i++) {
        sum = 0.0;
        for (j = 0; j < n; j++) {
            sum += A[i * n + j] * b[j];
        }
        c[i] = sum;
    }
}

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) {

    double *A, *b, *c;
    int i, j, Am, An;

    A = mxGetPr(prhs[0]);
    Am = (int)mxGetM(prhs[0]);   
    An = (int)mxGetN(prhs[0]); 

    c = malloc(Am * sizeof(double));
    b = mxGetPr(prhs[1]);

    mxv(Am, An, A, b, c); 

    for (i = 0; i < Am; i++)
        printf("c[%d] = %1.4f\n", i, c[i]);
 }

我使用以下输入调用 mex 函数:

A = [-865.6634 0 0 0;
    0 -17002.6822 0 0;
    0 0 -1726.2421 2539.6267;
    0 0 -2539.6267 -1726.2421;]
b = [-0.00153521; -0.00011165; -0.00037659; 0.00044981]

正确的结果应该是:

1.3290
1.8983
1.7924
0.1799

但是我明白了

c[0] = 1.3290
c[1] = 1.8983
c[2] = -0.4923
c[3] = -1.7329

所以前两个是正确的(c[0] 和 c[1])，但后两个不是。

我在我的代码中添加了一堆打印语句来尝试找出错误发生的位置:

#include "mex.h"

void mxv(int m, int n, double *A, double *b, double *c) {
    double sum;
    int i, j;

    for (i = 0; i < m; i++) {
        sum = 0.0;
        printf("********\n");
        for (j = 0; j < n; j++) {
            printf("A[%d][%d] = %1.10f\nb[%d] = %1.10f\n", i , j, A[i + n * j], j, b[j]);
            printf("A[%d][%d]*b[%d] = %1.10f\n", i, j, j, A[i * n + j] * b[j]);
            sum += A[i * n + j] * b[j];
            printf("sum = %1.10f\n", sum);
        }
        c[i] = sum;
    }
} 

void mexFunction(int nlhs, mxArray *plhs[],
                 int nrhs, const mxArray *prhs[]) {

    double *A, *b, *c;    
    int i, j, Am, An;

    A = mxGetPr(prhs[0]);
    Am = (int)mxGetM(prhs[0]);   
    An = (int)mxGetN(prhs[0]); 
    printf("size(A) = (%d,%d)\n", Am, An);
    for (i = 0; i < Am; i++) {
        for (j = 0; j < An; j++) {
            printf("A[%d][%d] = %1.4f\n", i, j, A[i + Am * j]);
        }
    }

    c = malloc(Am *sizeof(double));

    b = mxGetPr(prhs[1]);
    for (i = 0; i < Am; i++) {
        printf("b[%d] = %1.4f\n", i, b[i]);
    }
    mxv(Am, An, A, b, c);

    for (i = 0; i < Am; i++)
        printf("c[%d] = %1.4f\n", i, c[i]);    
}

我非常有信心从 Matlab 获得正确的输入:

size(A) = (4,4)
A[0][0] = -865.6634
A[0][1] = 0.0000
A[0][2] = 0.0000
A[0][3] = 0.0000
A[1][0] = 0.0000
A[1][1] = -17002.6822
A[1][2] = 0.0000
A[1][3] = 0.0000
A[2][0] = 0.0000
A[2][1] = 0.0000
A[2][2] = -1726.2421
A[2][3] = 2539.6267
A[3][0] = 0.0000
A[3][1] = 0.0000
A[3][2] = -2539.6267
A[3][3] = -1726.2421
b[0] = -0.0015
b[1] = -0.0001
b[2] = -0.0004
b[3] = 0.0004

但是当我研究矩阵 vector 乘法时，我发现乘法结果在某些情况下出于某种原因有错误的符号。这是它为 i = 2 打印的内容:

********
A[2][0] = 0.0000000000
b[0] = -0.0015352100
A[2][0]*b[0] = -0.0000000000
sum = 0.0000000000
A[2][1] = 0.0000000000
b[1] = -0.0001116500
A[2][1]*b[1] = -0.0000000000
sum = 0.0000000000
A[2][2] = -1726.2421000000
b[2] = -0.0003765900
A[2][2]*b[2] = 0.6500855124
sum = 0.6500855124
A[2][3] = 2539.6267000000
b[3] = 0.0004498100
A[2][3]*b[3] = -1.1423494859 <- THIS SHOULD BE 1.142349... (no minus sign)
sum = -0.4922639735
********

i=3 的情况类似。

提前致谢!

Answer 1

在:

printf("A[%d][%d] = %1.10f\nb[%d] = %1.10f\n", i , j, A[i + n * j], j, b[j]);
printf("A[%d][%d]*b[%d] = %1.10f\n", i, j, j, A[i * n + j] * b[j]);

第一个 printf 使用 A[i + n*j]，而第二个使用 A[i*n + j]。这些是数组中的转置位置。