MKL 函数实现矩阵每一行放大不同倍数

以下是使用 MKL 函数实现矩阵每一行放大不同倍数的示例代码：

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

int main()
{
    int nrows = 3;
    int ncols = 4;
    float scale[] = {1.0, 2.0, 3.0};
    float A[] = {1, 2, 3, 4,
                 5, 6, 7, 8,
                 9, 10, 11, 12};
    float *B = malloc(nrows * ncols * sizeof(float));
    int lda = ncols;
    int ldb = ncols;

    for (int i = 0; i < nrows; i++) {
        cblas_sscal(ncols, scale[i], A + i * lda, 1);
    }

    cblas_scopy(nrows * ncols, A, 1, B, 1);

    for (int i = 0; i < nrows; i++) {
        for (int j = 0; j < ncols; j++) {
            printf("%.1f ", B[i * ldb + j]);
        }
        printf("\n");
    }

    free(B);
    return 0;
}

在这个代码中，我们首先定义了一个 3 行 4 列的矩阵 A 和一个长度为 3 的数组 scale，数组中的每个元素表示对应行需要放大的倍数。然后，我们使用 cblas_sscal 函数对每一行进行放大操作，该函数的第一个参数 ncols 表示每一行的元素个数，第二个参数 scale[i] 表示第 i 行需要放大的倍数，第三个参数 A + i * lda 表示第 i 行的起始地址，最后一个参数 1 表示每隔一个元素进行一次操作。接着，我们使用 cblas_scopy 函数将矩阵 A 的内容复制到 B 中，最后打印出 B 中的每个元素，即可得到每一行放大后的结果。