aclnnUnfoldGrad

产品支持情况

[object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]。
[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]。

功能说明

算子功能：实现Unfold算子的反向功能，计算相应的梯度。

Unfold算子根据入参self，计算出维度 $dim$ 的所有大小为 $size$ 的切片。两个切片之间的步长由 $step$ 给出。如果 $sizedim$ 是入参self的维度 $dim$ 的大小，则返回的张量中维度 $dim$ 的大小将为 $(sizedim-size)/step+1$ 。返回的张量中附加了一个大小为 $size$ 的附加维度。

UnfoldGrad算子入参gradOut的shape为Unfold正向输出的shape，入参inputSizes为Unfold正向输入self的shape，UnfoldGrad算子出参gradIn的shape为Unfold正向入参self的shape。

例子：

[object Object]

函数原型

每个算子分为undefined，必须先调用“aclnnUnfoldGradGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnUnfoldGrad”接口执行计算。

aclnnStatus aclnnUnfoldGradGetWorkspaceSize(const aclTensor *gradOut, const aclIntArray *inputSizes, int dim, int size, int step, aclTensor *gradIn, uint64_t *workspaceSize, aclOpExecutor **executor)
aclnnStatus aclnnUnfoldGrad(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

aclnnUnfoldGradGetWorkspaceSize

参数说明：
- gradOut(aclTensor*, 计算输入)：Device侧的aclTensor，表示梯度更新系数，shape为(..., (sizedim-size)/step+1, size)，要求满足gradOut的第dim维等于 $(inputSizes[dim]-size)/step+1$ 和gradOut的size等于inputSizes的size+1。数据类型支持FLOAT、FLOAT16、BFLOAT16，undefined支持ND。
- inputSizes(aclIntArray*, 计算输入): Host侧的aclIntArray，表示输出张量的形状，值为(..., sizedim)，inputSizes的size小于等于8。数据类型支持INT64，undefined支持ND。
- dim（int64_t，计算输入）：公式中的 $dim$ 。表示展开发生的维度。 $dim$ 需要满足dim大于等于0且dim小于inputSizes的size。
- size（int64_t，计算输入）：公式中的 $size$ 。表示展开的每个切片的大小。 $size$ 需要满足size大于0且size小于等于inputSizes的第dim维。
- step（int64_t，计算输入）：公式中的 $step$ 。表示每个切片之间的步长。 $step$ 需要满足step大于0。
- gradIn（aclTensor*，计算输出）：表示Unfold的对应梯度，Device侧的aclTensor，shape为inputSizes。数据类型支持FLOAT、FLOAT16、BFLOAT16，且数据类型必须和gradOut一致。
- workspaceSize(uint64_t*, 出参)：返回需要在Device侧申请的workspace大小。
- executor(aclOpExecutor**, 出参)：返回op执行器，包含了算子计算流程。
返回值：

aclnnStatus：返回状态码，具体参见undefined。

[object Object]

aclnnUnfoldGrad

参数说明：
- workspace(void*, 入参)：在Device侧申请的workspace内存地址。
- workspaceSize(uint64_t, 入参)：在Device侧申请的workspace大小，由第一段接口aclnnUnfoldGradGetWorkspaceSize获取。
- executor(aclOpExecutor*, 入参)：op执行器，包含了算子计算流程。
- stream(aclrtStream, 入参)：指定执行任务的Stream。
返回值：

aclnnStatus：返回状态码，具体参见undefined。

约束说明

gradOut的shape满足约束： 1）gradOut的第dim维等于(inputSizes[dim]-size)/step+1 2）gradOut的size等于inputSizes的size+1
dim、size、step的要求： 1）dim大于等于0且dim小于inputSizes的size 2）size大于0且size小于等于inputSizes的第dim维 3）step大于0

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考undefined。

[object Object]