文件最后提交记录最后更新时间
1 个月前
1 个月前
1 个月前
1 个月前
26 天前
1 个月前
26 天前
README

CumulativeLogsumexp

产品支持情况

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

功能说明

  • 算子功能:对输入张量 x 沿 axis 指定维度计算累积 log-sum-exp,并将结果保存到输出张量 y 中。
  • 计算公式:设 iiaxis 维度上的下标,SiS_i 是参与第 ii 个输出元素计算的前缀或后缀下标集合,则:

yi=log⁡∑j∈Siexjy_i = \log \sum_{j \in S_i} e^{x_j}

exclusivefalse 时,SiS_i 包含当前位置;当 exclusivetrue 时,SiS_i 不包含当前位置。当 reversefalse 时按下标递增方向累积;当 reversetrue 时按下标递减方向累积。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入Tensor。shape支持1-6维,指定axis维度长度必须大于0。 FLOAT、FLOAT16 ND
axis 输入 需要进行累积log-sum-exp的维度,必须为编译期常量标量,取值范围为[-rank(x), rank(x)-1]。 INT32、INT64、INT16 ND
exclusive 属性 默认值为false。false表示包含当前位置;true表示不包含当前位置。 BOOL -
reverse 属性 默认值为false。false表示正向累积;true表示反向累积。 BOOL -
y 输出 输出Tensor,shape与x相同。 FLOAT、FLOAT16 ND

约束说明

  • axis 必须为常量标量。
  • x 不支持标量输入;axis 指定维度长度为0时返回参数错误。
  • y 的shape和数据类型需要与 x 一致。

调用说明

调用方式 样例代码 说明
图模式调用 test_geir_cumulative_logsumexp.cpp 通过算子IR构图方式调用CumulativeLogsumexp算子。