Question

我有一个nxd矩阵V=[v_1; v_2;...; v_n]（;表示新行），其中v_i是1xd个向量。

我想计算以下总和：v_1^T*v_1 + v_2^T*v_2 + ... + v_n^T*v_n，这是一个dxd矩阵（v_i^T是v_i)的转置。

目前我使用for循环，如下面的代码所示，当n非常大时（我认为如此），效率不高。

#include <iostream>
#include <opencv2/core.hpp>
using namespace cv;
using namespace std;

int main (int argc, char * argv[])
{   
    int n=5, d=3;

    Mat V = Mat(n, d, CV_32F);
    randu(V, Scalar::all(0), Scalar::all(10));        
    cout<<V<<endl<<endl;

    Mat M = Mat::zeros(d, d, CV_32F);
    for(int i=0; i<n; i++)
    {
        M = M + V.row(i).t()*V.row(i);
    }

    cout<<M<<endl<<endl;    
    return 0;
}

希望有人可以建议更快的方式。提前谢谢。

Answer 1

你可以拿V.t（）* V

（我花了一分钟才意识到这一点，但如果你通过矩阵乘法，你会看到它是相同的）

OpenCV：快速矩阵计算

1 个答案: