ForeachAddcmulScalar

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×
Kirin X90 处理器系列产品	√
Kirin 9030 处理器系列产品	√

功能说明

• 算子功能：先对张量列表x2和张量列表x3执行逐元素乘法，再乘以张量scalar，最后将之前计算的结果与张量列表x1执行逐元素相加。

• 计算公式：

  $$\begin{gathered} x1=[x{1}_0,x{1}_1,...x{1}_{n-1}],x2=[x{2}_0,x{2}_1,...x{2}_{n-1}],x3=[x{3}_0,x{3}_1,...x{3}_{n-1}] \\ y=[y_0,y_1,...y_{n-1}] \end{gathered}$$

  $${\mathrm{y}}_i=x{1}_i+\mathrm{s}\mathrm{c}\mathrm{a}\mathrm{l}\mathrm{a}\mathrm{r}\times x{2}_i\times x{3}_i(i=0,1,...n-1)$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x1	输入	表示混合运算中加法的第一个输入张量列表，对应公式中的`x1`。该参数中所有Tensor的数据类型保持一致。	FLOAT32、FLOAT16、INT32、BFLOAT16	ND
x2	输入	表示混合运算中乘法的第二个输入张量列表，对应公式中的`x2`。数据类型、数据格式和shape与入参`x1`的数据类型、数据格式和shape一致，该参数中所有Tensor的数据类型保持一致。	FLOAT32、FLOAT16、INT32、BFLOAT16	ND
x3	输入	表示混合运算中乘法的第三个输入张量列表，对应公式中的`x3`。数据类型、数据格式和shape与入参`x1`的数据类型、数据格式和shape一致，该参数中所有Tensor的数据类型保持一致。	FLOAT32、FLOAT16、INT32、BFLOAT16	ND
scalar	输入	表示混合运算中乘法的第一个输入张量，对应公式中的`scalar`。元素个数为1。	FLOAT32、FLOAT16、INT32	ND
y	输出	表示混合运算的输出张量列表，对应公式中的`y`。数据类型和数据格式与入参`x1`的数据类型和数据格式一致，shape size大于等于入参`x1`的shape size。该参数中所有Tensor的数据类型保持一致。	FLOAT32、FLOAT16、INT32、BFLOAT16	ND

• Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品：

  • 当x1的数据类型为FLOAT32、FLOAT16、INT32时，scalar数据类型与x1的数据类型保持一致。
  • 当x1的数据类型为BFLOAT16时，scalar数据类型支持FLOAT32。

• Ascend 950PR/Ascend 950DT：

  • 当x1的数据类型为FLOAT32、INT32时，scalar数据类型与x1的数据类型保持一致。
  • 当x1的数据类型为BFLOAT16时，scalar数据类型支持FLOAT32。
  • 当x1的数据类型为FLOAT16时，scalar数据类型支持FLOAT16, FLOAT32。

• Kirin X90/Kirin 9030处理器系列产品：不支持BFLOAT16。

约束说明

• 输出不支持非连续Tensor。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_foreach_addcmul_scalar	通过aclnnForeachAddcmulScalar接口方式调用ForeachAddcmulScalar算子。
aclnn接口	test_aclnn_foreach_addcmul_scalar_v2	通过aclnnForeachAddcmulScalarV2接口方式调用ForeachAddcmulScalar算子。
图模式	-	通过算子IR构图方式调用ForeachAddcmulScalar算子。