accumulate - 2023.2 日本語

AI エンジン ML カーネルおよびグラフ プログラミング ガイド (UG1603)

Document ID
UG1603
Release Date
2023-12-07
Version
2023.2 日本語

AI エンジン API は、係数と乗算する可変数のベクターの累算をサポートします (aie::accumulate)。次の図に、4 つのベクターの重み累算の例を示します。

図 1. ベクター算術演算

次に、aie::accumulate のコード例を示します。

/*template<unsigned Lanes, AccumOrOp Acc, Vector VecCoeff, Vector VecData, Vector... NextVecData>
aie::accumulate(const Acc & acc,
    const VecCoeff & coeff,
    unsigned coeff_start,
    const VecData & data,
    const NextVecData &... next_data)
*/
aie::vector<int16,16> coe;
aie::vector<int16,16> vdata,vnext1,vnext2,vnext3,vnext4;
//acc[i]=coe[0]*vdata[i]+coe[1]*vnext1[i]+coe[2]*vnext2[i]+coe[3]*vnext3[i]
auto acc=aie::accumulate<16>(coe,2,vdata,vnext1,vnext2,vnext3);//coeff_start=2
acc=aie::accumulate<16>(acc,coe,2,vdata,vnext4);//coeff_start=2
注記: すべてのベクターのデータ型とサイズは同じである必要があります。