aclnnIm2colBackward

产品支持情况

[object Object]undefined

功能说明

• 算子功能：从批处理输入张量中提取滑动局部块，将滑动局部块数组合并为一个大张量。

• 计算公式：

  考虑一个形状为 $(N,C,*)$的批处理input张量，其中$N$是批处理维度，$C$是通道维度，而$*$表示任意空间维度。

  此操作将input空间维度内的每个滑动kernel_size大小的块展平为形状是$(N,C×\prod(kernel\_size),L)$ 的 3-D output张量的列（即最后一维）。

  其中：

  • $C×\prod(kernel\_size)$ 是每个块内的值的数量（一个块有$\prod(kernel\_size)$ 个空间位置，每个空间位置都包含一个$C$ 通道向量），而$L$是这些块的总数：

    $$L=\prod_d⌊{\frac{spatial\_size[d]+2×padding[d]−dilation[d]×(kernel\_size[d]−1)−1}{stride[d]}+1}⌋$$

  • spatial_size由input(上面的$*$)的空间维度构成，而$d$覆盖所有空间维度。 因此，在最后一个维度（列维度）索引，output会给出某个块内的所有值。

函数原型

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

aclnnIm2colBackwardGetWorkspaceSize

• 参数说明：

  [object Object]

• 返回值：

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

  第一段接口完成入参校验，出现以下场景时报错：

  [object Object]

aclnnIm2colBackward

• 参数说明：

  [object Object]

• 返回值：

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

约束说明

无。

调用示例

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