Addr
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | × |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | × |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | × |
| Atlas 训练系列产品 | × |
功能说明
-
算子功能:求一维向量vec1和vec2的外积得到一个二维矩阵,并将外积结果矩阵乘一个系数后和自身乘系数相加后输出
-
计算公式:
out=β self+α (vec1⊗vec2)\text{out} = \beta\ \text{self} + \alpha\ (\text{vec1} \otimes\text{vec2})
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x1 | 输入 | 待进行addr计算的入参,公式中的self。 | FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、BOOL | ND |
| x2 | 输入 | 待进行addr计算的入参,公式中的vec1。 | FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、BOOL | ND |
| x3 | 输入 | 待进行addr计算的入参,公式中的vec2。 | FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、BOOL | ND |
| beta | 输入 | 待进行addr计算的入参,公式中的β。 | FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、BOOL | ND |
| alpha | 输入 | 待进行addr计算的入参,公式中的α。 | FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、BOOL | ND |
| y | 输出 | 待进行addr计算的出参,公式中的out。 | FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、BOOL | ND |
- Atlas 训练系列产品:数据类型支持FLOAT、FLOAT16、DOUBLE、INT8、INT16、INT32、INT64、UINT8、BOOL。
- Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品、Ascend 950PR/Ascend 950DT:数据类型支持FLOAT、FLOAT16、DOUBLE、INT8、INT16、INT32、INT64、UINT8、BOOL、BFLOAT16。
约束说明
无
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_addr | 通过aclnnAddr接口方式调用Addr算子。 |