AxpyV2

产品支持情况

产品	是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√

功能说明

• 算子功能：源操作数2(src2Tensor)中每个元素与标量(alphaScalar)对应元素求积后和源操作数1(src1Tensor)中的对应元素相加。

• 计算公式：

$$dstTenso{r}_i=src1Tenso{r}_i+alphaScalar\ast src2Tenso{r}_i$$

参数说明

参数名	输入/输出	描述	使用说明	数据类型	数据格式	维度(shape)	非连续Tensor
self	输入	待进行axpy_v2计算的入参，公式中的src1Tensor。	无	FLOAT、FLOAT16、BFLOAT16、INT32	ND	0-8	√
other	输入	待进行axpy_v2计算的入参，公式中的src2Tensor。	shape与x1相同。	FLOAT、FLOAT16、BFLOAT16、INT32	ND	0-8	√
alpha	输入	待进行axpy_v2计算的入参，公式中的alphaScalar。	shape为[]。	FLOAT、FLOAT16、BFLOAT16、INT32	ND	0-8	√
out	输出	待进行axpy_v2计算的出参，公式中的dstTensor。	shape与x1相同。	FLOAT、FLOAT16、BFLOAT16、INT32	ND	0-8	√

## 约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_axpy_v2	通过aclnnAxpyV2接口方式调用AxpyV2算子。

贡献说明

贡献者	贡献方	贡献算子	贡献时间	贡献内容
Nice_try	个人开发者	AxpyV2	2025/11/25	AxpyV2算子适配开源仓