文档
注册

GetAccVal

功能说明

获取ReduceSum(针对tensor前n个数据计算)接口的计算结果。

函数原型

template <typename T>
__aicore__ inline T GetAccVal();

参数说明

表1 模板参数说明

参数名

描述

T

ReduceSum指令的数据类型,支持half/float。

返回值

ReduceSum(针对tensor前n个数据计算)接口的计算结果。

支持的型号

Atlas A2训练系列产品/Atlas 800I A2推理产品

约束说明

无。

调用示例

// LocalTensor<float> src;
// LocalTensor<float> work;
// LocalTensor<float> dst;
ReduceSum(dst, src, work, 128);
float res = GetAccVal<float>();
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词