FilesLast commitLast update
13 days ago
2 months ago
14 days ago
14 days ago
2 months ago
3 months ago
4 months ago
15 days ago
README

LogAddExp

产品支持情况

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

功能说明

  • 算子功能:对两个输入张量逐元素计算指数和的对数,默认属性下以e为底。
  • 计算公式:

    y=ln⁡(ex1+ex2)y = \ln\left(e^{x_1} + e^{x_2}\right)

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x1 输入 公式中的输入张量x_1 FLOAT16, BFLOAT16, FLOAT32 ND
x2 输入 公式中的输入张量x2 同x1 ND
base 可选属性 指数和对数运算的底数,默认值为-1.0,表示以e为底 FLOAT32 -
scale 可选属性 指数项中-abs(x1-x2)的缩放系数,默认值为1.0 FLOAT32 -
shift 可选属性 指数项的偏移量,默认值为0.0 FLOAT32 -
y 输出 公式中的输出张量y 同x1 ND

约束说明

  • base必须为-1.0或大于0的值。
  • x1、x2、y的数据类型需相同。
  • x1与x2的shape需满足broadcast关系,y的shape为broadcast后的shape。
  • Atlas推理系列产品、Atlas训练系列产品不支持BFLOAT16数据类型。

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_log_add_exp 通过aclnnLogAddExp接口方式调用LogAddExp算子
aclnn调用 test_aclnn_log_add_exp2 通过aclnnLogAddExp2接口方式调用LogAddExp算子