C:矩阵向量乘积,乘以两个双数给出错误的符号

问题描述 投票:3回答:1

我正在尝试执行一个简单的矩阵时间向量乘法,由于某些原因,我在我的结果中得到了错误的符号。我不知道为什么会这样,任何指针都会非常感激。

这是我的整个代码,即矩阵*向量函数和来自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]),而不是后两个。

我在我的代码中添加了一堆print语句,试图找出错误发生的位置:

#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

但是当我查看矩阵向量乘法时,我发现乘法结果在某些情况下由于某种原因出现了错误的符号。这是它为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的情况,会发生类似的情况。

提前致谢!

c double matrix-multiplication mex
1个回答
4
投票

在:

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]。这些是阵列中的转置位置。

© www.soinside.com 2019 - 2024. All rights reserved.