AddMatMatElements
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
功能说明
-
算子功能：对输入张量
a、b进行逐元素相乘后，与输入张量c按标量alpha、beta进行加权求和，结果写入输出张量cOut。 -
计算公式：
cOuti=ci×β+α×ai×bicOut_i = c_i \times \beta + \alpha \times a_i \times b_i
其中
a、b、c、cOut具有相同的 shape 和 dtype，alpha、beta为标量。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| a | 输入 | 公式中的输入张量 a，参与逐元素相乘。 | FLOAT16、FLOAT、BFLOAT16 | ND |
| b | 输入 | 公式中的输入张量 b，与 a 同 shape、同 dtype。 | 同 a | ND |
| c | 输入 | 公式中的输入张量 c，与 a 同 shape、同 dtype。 | 同 a | ND |
| alpha | 输入（aclScalar） | 标量缩放系数，对 a × b 的乘积进行缩放。 | FLOAT | - |
| beta | 输入（aclScalar） | 标量缩放系数，对 c 进行缩放。 | FLOAT | - |
| cOut | 输出 | 公式中的输出张量 cOut，与 a 同 shape、同 dtype。 | 同 a | ND |
约束说明
a、b、c、cOut四个张量必须具有完全一致的 shape 和 dtype。- 张量最大维度为 8。
- 仅支持 ND 格式，不支持私有格式。
alpha、beta不可为空指针。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_add_mat_mat_elements | 通过 aclnnAddMatMatElements 两段式接口调用 AddMatMatElements 算子，覆盖 FP16 / FP32 / BF16 三种 dtype |