您可以验证此 C 算法是否执行了小矩阵的正确求逆:
gcc main.c -llapacke -llapack
dac@dac-Latitude-E7450 ~/C/gnu> ./a.out
dgetrf eh, 0, should be zero
dgetri eh, 0, should be zero
0.6, -0.7
-0.2, 0.4⏎
上面的数字是针对这个示例程序中的 2*2 矩阵进行验证的:
#include <stdio.h>
#include <stddef.h>
#include <lapacke.h>
int main() {
int N = 2;
int NN = 4;
double M[2][2] = {{4, 7},
{2, 6}};
int pivotArray[2];
int errorHandler;
double lapackWorkspace[4];
dgetrf_(&N, &N, M[0], &N, pivotArray, &errorHandler);
printf("dgetrf eh, %d, should be zero\n", errorHandler);
dgetri_(&N, M[0], &N, pivotArray, lapackWorkspace, &NN, &errorHandler);
printf("dgetri eh, %d, should be zero\n", errorHandler);
for (size_t row = 0; row < N; ++row) {
for (size_t col = 0; col < N; ++col) {
printf("%g", M[row][col]);
if (N - 1 != col) { printf(", "); }
}
if (N - 1 != row) { printf("\n"); }
}
return 0;
}
现在我们只需要定义一个更大的矩阵 1024*1024 并以同样的方式反转它。
#include <stdio.h>
#include <lapacke.h>
int main() {
int N = 1024;
int NN = N*N;
double M[N][N];
for(int i=0;i<N;i++) {
for(int j=0;j<N;j++) {
M[i][j] = 0;
if(i==j)
M[i][j] = 1;
}
}
int pivotArray[N];
int errorHandler;
double lapackWorkspace[N*N];
dgetrf_(&N, &N, M[0], &N, pivotArray, &errorHandler);
printf ("dgetrf eh, %d, should be zero\n", errorHandler);
dgetri_(&N, M[0], &N, pivotArray, lapackWorkspace, &NN, &errorHandler);
printf("dgetri eh, %d should be zero\n", errorHandler);
return 0;
}
要运行上述代码,我还必须在 Linux 上增加堆栈大小:
ulimit -s 65532
上面代码使用的矩阵是单位矩阵,它是它自己的逆矩阵。您还可以使用具有逆矩阵的任何其他矩阵,并将其反转两次以检查您是否得到原始矩阵。