Axpy
函数功能
源操作数(srcLocal)中每个元素与标量求积后和目的操作数(dstLocal)中的对应元素相加，计算公式如下，其中PAR表示矢量计算单元一个迭代能够处理的元素个数：
函数原型
tensor前n个数据计算：
template <typename T, typename U>
__aicore__ inline void Axpy(const LocalTensor<T>& dstLocal, const LocalTensor<U>& srcLocal, const U& scalarValue, const int32_t& calCount)
参数说明
表1 模板参数说明
| 参数名 | 描述 |
|---|---|
| T | 目的操作数数据类型。 |
| U | 源操作数数据类型。 |
表2 参数说明
| 参数名称 | 类型 | 说明 |
|---|---|---|
| dstLocal | 输出 | 目的操作数。 类型为LocalTensor，支持的TPosition为VECIN/VECCALC/VECOUT。 LocalTensor的起始地址需要32字节对齐。 Kirin9020训练系列产品，支持的数据类型为：half、float |
| srcLocal | 输入 | 源操作数。 类型为LocalTensor，支持的TPosition为VECIN/VECCALC/VECOUT。 LocalTensor的起始地址需要32字节对齐。 Kirin9020训练系列产品，支持的数据类型为：half、float |
| scalarValue | 输入 | 源操作数，scalar标量。支持的数据类型为：half/float。scalarValue的数据类型需要和srcLocal保持一致。 |
| calCount | 输入 | 输入数据元素个数。 |
返回值
无
支持的型号
Kirin9020系列处理器
KirinX90系列处理器
约束说明
该接口支持的精度组合如下。
- half精度组合：srcLocal数据类型=half；scalar数据类型=half；dstLocal数据类型=half；PAR=128。
- float精度组合：srcLocal数据类型=float；scalar数据类型=float；dstLocal数据类型=float；PAR=64。
- mixed精度组合：srcLocal数据类型=half；scalar数据类型=half；dstLocal数据类型=float；PAR=64。
调用示例
本样例中只展示Compute流程中的部分代码。如果开发者需要运行样例代码，请将该代码段拷贝并替换更多样例完整样例模板中Compute函数的部分代码即可。
tensor前n个数据计算样例：
AscendC::Axpy(dstLocal, src0Local, (half)2.0, 512);// half精度组合