算子功能:在3d的逆运算,由outputSize决定outRef的D、H、W轴大小,并根据indices索引在outRef中填入self的元素值,其余位置都设置为0。
计算公式:
输入为4维,各维度分别为N、D、H、W,(其中N(Batch)表示批量大小、H(Height)表示特征图高度、W(Width)表示特征图宽度、D(Depth)表示特征图深度)时:
输入为5维,各维度分别为N、C、D、H、W,(其中C(Channels)表示特征图通道)时:
其中outRef、indices和self是最后两轴合为一轴,经过reshape得到的,i ∈ [0, D * H * W)。
- 每个算子分为,必须先调用“aclnnMaxUnpool3dGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnMaxUnpool3d”接口执行计算。
[object Object]
[object Object]
- 确定性计算:
- aclnnMaxUnpool3d默认确定性实现。
[object Object]