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);
}