GetReduceRepeatSumSpr(ISASI)
产 品 支 持 情 况
功 能 说 明
获 取ReduceSum接 口(Tensor前n个 数 据 计 算 接 口,n为 接 口 的count参 数)的 计 算 结 果。
函 数 原 型
Text
template <typename T>
__aicore__ inline T GetReduceRepeatSumSpr()
参 数 说 明
表 1 模 板 参 数 说 明
返 回 值 说 明
ReduceSum接 口(Tensor前n个 数 据 计 算 接 口,n为 接 口 的count参 数)的 计 算 结 果。
约 束 说 明
无。
调 用 示 例
具 体 可 参 考GetReduceRepeatSumSpr样 例。
Text
AscendC::LocalTensor<float> src;
AscendC::LocalTensor<float> work;
AscendC::LocalTensor<float> dst;
AscendC::ReduceSum(dst, src, work, 128);
float res = AscendC::GetReduceRepeatSumSpr<float>(); // 返 回 前128个 数 的 和