gpt4 book ai didi

c++ - LAPACK 矩阵乘法与 C++

转载 作者:太空狗 更新时间:2023-10-29 21:46:09 25 4
gpt4 key购买 nike

我是 C++ 的新手,正在尝试使用 LAPACK 库进行矩阵乘法。我尝试运行下面给出的例程 dgemm。我期待输出 A*B。但每次我得到的答案都是 B*A。这是例程的工作方式还是我的代码有任何问题。

我的代码:

#include "stdafx.h"
#include<iostream>

using namespace std;

extern "C" void dgemm_(const char *TRANSA, const char *TRANSB, const int *M, const int *N, const int *K, double *ALPHA, double *A, const int *LDA, double *B, const int *LDB, double *BETA, double *C, const int *LDC);

int main(void)
{
double A[4] = {1,2,3,4};
double B[4] = {5,6,7,8};
char TRANS = 'N';
int M = 2;
int N = 2;
int K = 2;
double ALPHA = 1.0;
int LDA = 2;
int LDB = 2;
double BETA = 0.0;
double C[4];
int LDC = 2;

dgemm_(&TRANS, &TRANS, &M, &N, &K, &ALPHA, A, &LDA, B, &LDB, &BETA, C, &LDC);

cout << C[0] << endl;
cout << C[1] << endl;
cout << C[2] << endl;
cout << C[3] << endl;
getchar();
return 0;
}

任何输入都会非常有帮助。

最佳答案

我没有研究你调用 dgemm 的细节,我不知道你是如何解释结果矩阵的。但是您似乎很可能在某处混淆了 col major 和 row major 的解释。计算很可能使用 col major,但您假设是 row major。

关于c++ - LAPACK 矩阵乘法与 C++,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/15830913/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com