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

任何建议，参考，请。

Answer 1

ArrayFire 不提供双点积功能，但您可以定义自己的。

假设双点积的定义与此处和此处给出的一致，“双点积”是计算两个矩阵的元素乘法后所有值的总和。两个矩阵必须具有相同的形状：

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