7

我有两个矩阵:A 和 B。

  1. 我该如何存储它们?
  2. 如何使用 Accelerate 框架计算矩阵 A 的逆矩阵?
  3. 如何找到 A*B 的乘积?
  4. 如何使用 Accelerate 框架转置矩阵 A?

谢谢你回答我的问题!

帮助文件

#import <Foundation/Foundation.h>
#include <Accelerate/Accelerate.h>

@interface Working_with_matrices : NSObject
-(int)invert_matrix:(int) N andWithMatrix:(double*) matrix;
@end

实施文件

#import "Working_with_matrices.h"
#include <Accelerate/Accelerate.h>

@implementation Working_with_matrices
-(int) matrix_invert:(int) N andWithMatrix:(double*)matrix
{    
int error=0;
int *pivot = malloc(N*N*sizeof(int));
double *workspace = malloc(N*sizeof(double));

dgetrf_(&N, &N, matrix, &N, pivot, &error);

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

dgetri_(&N, matrix, &N, pivot, workspace, &N, &error);

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

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

从主函数调用我的代码

#import <Foundation/Foundation.h>
#import "Working_with_matrices.h"

int main(int argc, const char * argv[])
{
int N = 3;
double A[9];
Working_with_matrices* wm=[[Working_with_matrices alloc]init];

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;
[wm invert_matrix:N andWithMatrix:A];
//        [ -1.25  -1.0   3.25 ]
// A^-1 = [  0.5    1.0  -1.5  ]
//        [  0.25   0.0  -0.25 ] 
for (int i=0; i<9; i++) 
{
    NSLog(@"%f", A[i]);
}
return 0;
}
4

1 回答 1

8

我对使用加速框架还是有点陌生​​,但我会回答我能回答的问题。

  1. 加速框架期望矩阵作为一维数组传入。因此,如果您有一个 4x4 矩阵,则第一行将放置在数组的索引 0-3 中,第二行将放置在索引 4-7 中,依此类推。
  2. 我从来没有做过,但这个答案看起来是一个很好的起点。https://stackoverflow.com/a/11321499/385017
  3. 您要使用的方法是vDSP_mmul单精度或vDSP_mmulD双精度。您可能想查看它的文档以更好地了解如何使用它,但这里有一个示例可以帮助您入门。

    float *matrixA;  //set by you
    float *matrixB;  //set by you
    float *matrixAB; //the matrix that the answer will be stored in
    
    vDSP_mmul( matrixA, 1, matrixB, 1, matrixAB, 1, 4, 4, 4 );
    // the 1s should be left alone in most situations
    // The 4s in order are:
    //     the number of rows in matrix A
    //     the number of columns in matrix B
    //     the number of columns in matrix A and the number of rows in matrix B.
    
于 2012-08-14T18:07:46.343 回答