获取ReduceSum(针对tensor前n个数据计算)接口的计算结果。
1 2 | template <typename T> __aicore__ inline T GetAccVal() |
参数名 |
描述 |
---|---|
T |
ReduceSum指令的数据类型,支持half/float。 |
ReduceSum(针对tensor前n个数据计算)接口的计算结果。
Atlas A2训练系列产品/Atlas 800I A2推理产品
无。
1 2 3 4 5 | AscendC::LocalTensor<float> src; AscendC::LocalTensor<float> work; AscendC::LocalTensor<float> dst; AscendC::ReduceSum(dst, src, work, 128); float res = AscendC::GetAccVal<float>(); |