次元 N および m のインデックス (p,i) を持つ device_vector に保存された 2 次元配列 $a,$ があります。
計算したい
$$s_{ij} = \sum \limits_{p=1}^{N} a_{p,i} a_{p,j}$$
for $i,j=1,...,m.$
スラストを使用してこれを行う簡単な方法はありますか?
上記のコードはラテックスです。C++では、次のようになります
Matrix A(N,m); // filled with data
Matrix S(m,m);
for (int i=0; i <m;++i)
for (int j=0; j <m;++j)
{
S(i,j)=0;
for (int p=0; p < N; ++p)
S(i,j) += A(p,i)*A(p,j);
}