接口功能:
[object Object]函数用于计算输入张量在指定维度上的累积乘积。例如,如果有一个张量表示一系列的数值,[object Object]可以计算出这些数值从开始位置到当前位置的乘积序列。计算公式:
一维张量(向量)情况 当对于一维张量,累积乘积的计算公式为:
...
用数学公式表示。
高维张量情况(以二维张量为例, dim=0 沿行方向) 对于二维张量:
计算后的结果张量:
对于第一列(j=1):
所以对于任意列j,也有类似规律, 即:
高维张量情况(以二维张量为例, dim=1 沿列方向情况) 所以对于任意列j,也有类似规律, 即:
其它参数可以类似地根据上述规则进行推导
aclnnCumprod和aclnnInplaceCumprod实现相同的功能,使用区别如下,请根据自身实际场景选择合适的算子。
- aclnnCumprod:需新建一个输出张量对象存储计算结果。
- aclnnInplaceCumprod:无需新建输出张量对象,直接在输入张量的内存中存储计算结果。
每个算子分为,必须先调用"aclnnCumprodGetWorkspaceSize"或者"aclnnInplaceCumprodGetWorkspaceSize"接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用"aclnnCumprod"或者"aclnnInplaceCumprod"接口执行计算。
[object Object]
[object Object]
[object Object]
[object Object]
- 确定性计算:
- aclnnCumprod&aclnnInplaceCumprod默认确定性实现。
[object Object]