2

我有 2 个二维 NSMutableArrays，我正在尝试做一些基本的矩阵乘法。我在下面有我的通用公式代码，但它的性能非常慢（如预期的那样）。我做了很多谷歌搜索，但没有找到任何简单或易于理解的公式来更改代码以提高性能。谁能指出一个简单的公式/教程/示例的正确方向，说明如何在 Objective C 中使用矩阵乘法获得比 0(n^3) 更好的性能。

+ (NSMutableArray*)multiply:(NSMutableArray*)a1 withArray:(NSMutableArray*)a2
{
    if([[a1 objectAtIndex: 0] count] != [a2 count])
    {
        NSLog(@"Multiplicaton error!");
        return NULL;
    }

    int a1_rowNum = [a1 count];
    int a2_rowNum = [a2 count];
    int a2_colNum = [[a2 objectAtIndex:0] count];
    NSMutableArray *result = [NSMutableArray arrayWithCapacity:a1_rowNum];
    for (int i = 0; i < a1_rowNum; i++) {
        NSMutableArray *tempRow = [NSMutableArray arrayWithCapacity:a2_colNum];
        for (int j = 0; j < a2_colNum; j++) {
            double tempTotal = 0;
            for (int k = 0; k < a2_rowNum; k++) {
                double temp1 = [[[a1 objectAtIndex:i] objectAtIndex:k] doubleValue];
                double temp2 = [[[a2 objectAtIndex:k] objectAtIndex:j] doubleValue];
                tempTotal += temp1 * temp2;
            }
             //Stored as a string because I upload it to an online database for storage.
            [tempRow addObject:[NSString stringWithFormat:@"%f",tempTotal]];
        }
        [result addObject:tempRow];
    }
    return result;
}
4

2 回答 2

8

如果你用 C 编写它会快得多。

double[]与这个任务NSArray的s相比，速度快得离谱。NSNumber您将拥有良好的缓存一致性、最少的指令，无需通过运行时或分配来写入或读取元素。无需对每个元素执行引用计数循环……</p>

于 2012-05-22T05:15:18.107 回答
4

您需要查看Apple 的 Accelerate frameWork for ios4.0 onwards。你可以用它做很多复杂的数学和矩阵操作，这个框架经过优化，可以在任何 iOS 硬件上运行。

查看：

https://developer.apple.com/performance/accelerateframework.html

于 2012-05-22T05:16:35.873 回答