MKL 函数实现对角矩阵乘普通矩阵并保存结果

以下是使用 mkl 函数实现对角矩阵 A 乘普通矩阵 B 并将结果保存在矩阵 B 中的示例代码：

#include <stdio.h>
#include <stdlib.h>
#include "mkl.h"

int main()
{
    int n = 3;  // 矩阵维度
    double *A = (double*) malloc(n * n * sizeof(double));  // 对角矩阵 A
    double *B = (double*) malloc(n * n * sizeof(double));  // 普通矩阵 B

    // 初始化矩阵 A 和矩阵 B
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            A[i*n+j] = (i == j) ? i+1 : 0;  // 对角线元素为 i+1，其余为 0
            B[i*n+j] = i+j;  // 元素为 i+j
        }
    }

    // 打印矩阵 A 和矩阵 B
    printf("Matrix A:\n");
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            printf("%f ", A[i*n+j]);
        }
        printf("\n");
    }
    printf("Matrix B:\n");
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            printf("%f ", B[i*n+j]);
        }
        printf("\n");
    }

    // 使用 mkl 函数计算 A*B 并将结果保存在 B 中
    cblas_dgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, n, n, n, 1.0, A, n, B, n, 0.0, B, n);

    // 打印矩阵 B 的结果
    printf("Result:\n");
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            printf("%f ", B[i*n+j]);
        }
        printf("\n");
    }

    free(A);
    free(B);
    return 0;
}

在本示例中，我们使用 mkl 的 cblas_dgemm 函数来计算 A*B，其中 CblasNoTrans 表示不需要对 A 和 B 进行转置，1.0 表示矩阵 A 的系数，0.0 表示矩阵 B 的系数。最后的结果将保存在矩阵 B 中。