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算子。