如何使用加速框架执行矩阵逆运算?

时间:2012-07-01 14:25:30

标签: osx-lion lapack accelerate-framework matrix-inverse

我想找到矩阵的逆矩阵。

我知道这涉及到第一次LU因子化然后是反转步骤但是我找不到所需的功能,通过搜索苹果的10.7文档!

这似乎是一个有用的帖子Symmetric Matrix Inversion in C using CBLAS/LAPACK,指出应该使用sgetrf_sgetri_函数。但是,在搜索这些术语时,我在Xcode文档中找不到任何内容。

是否有人为此矩阵操作设置了锅炉铭牌代码?

1 个答案:

答案 0 :(得分:14)

Apple根本没有记录LAPACK代码,我猜是因为他们只是实现了netlib.org的标准接口。遗憾的是,您无法从内置的Xcode文档中搜索这些函数名称,但解决方案非常简单:只需在URL中指定函数名称即可。 dgetrf_()转到http://www.netlib.org/clapack/what/double/dgetrf.c

要反转矩阵,需要两个LAPACK函数:dgetrf_()执行LU分解,dgetri_()获取前一个函数的输出并执行实际反转。

我使用Xcode创建了一个标准的Application Project,添加了Accelerate Framework,创建了两个C文件:matinv.h,matinv.c并编辑了main.m文件以删除Cocoa的东西:

// main.m

#import "matinv.h"

int main(int argc, char *argv[])
{
    int N = 3;
    double A[N*N];
    A[0] = 1; A[1] = 1; A[2] = 7;
    A[3] = 1; A[4] = 2; A[5] = 1;
    A[6] = 1; A[7] = 1; A[8] = 3;
    matrix_invert(N, A);
    //        [ -1.25  -1.0  3.25 ]
    // A^-1 = [  0.5       1.0  -1.5  ]
    //        [  0.25   0.0 -0.25 ] 
    return 0;
}

现在是头文件,

//  matinv.h

int matrix_invert(int N, double *matrix);

然后是源文件,

int matrix_invert(int N, double *matrix) {

    int error=0;
    int *pivot = malloc(N*sizeof(int)); // LAPACK requires MIN(M,N), here M==N, so N will do fine.
    double *workspace = malloc(N*sizeof(double));

    /*  LU factorisation */
    dgetrf_(&N, &N, matrix, &N, pivot, &error);

    if (error != 0) {
        NSLog(@"Error 1");
        free(pivot);
        free(workspace);
        return error;
    }

    /*  matrix inversion */
    dgetri_(&N, matrix, &N, pivot, workspace, &N, &error);

    if (error != 0) {
        NSLog(@"Error 2");
        free(pivot);
        free(workspace);
        return error;
    }

    free(pivot);
    free(workspace);
    return error;
}