c++ - Arrayfire のダブル ドット プロダクト

Question

Arrayfireで二重内積を計算したい。二重ドット積に使用するarrayfireの関数はありますか。これは、私が実行したい私のテンソルです:

AA:(E_iE_i - c^2I)
where, A--> 2D-array ; E_i --> 1D array (defined as col. vec in arrayfire)
c --> scalar/constant ; I --> identity matrix

任意の提案、参照、plz.

Answer 1

ArrayFire は二重内積関数を提供しませんが、独自に定義できます。

こことここで与えられたものと一致する二重内積の定義を仮定すると、「二重内積」は、2 つの行列の要素ごとの乗算を計算した後のすべての値の合計です。2 つの行列は同じ形状でなければなりません。

af::array doubledot(af::array a, af::array b) {
  return af::sum(a * b);
}