Col2im

产品支持情况

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

功能说明

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

• 计算公式：

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

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

  其中：

  • $C\times \prod (kerne{l}_{s}ize)$是每个块内的值的数量（一个块有$\prod (kerne{l}_{s}ize)$个空间位置，每个空间位置都包含一个$C$通道向量），而$L$是这些块的总数：

    $$L=\prod _d\lfloor \frac{spatial\_size[d]+2\times padding[d]-dilation[d]\times (kernel\_size[d]-1)-1}{stride[d]}+1\rfloor$$

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

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
gradOutput	输入	反向输入，对应公式中的output。	BFLOAT16、FLOAT16、FLOAT	5HD
inputSize	属性	输入张量的形状，对应公式中的spatial_size。	INT64	ND
kernelSize	属性	卷积核的大小，对应公式中的kernel_size。	INT64	ND
dilation	属性	膨胀参数，对应公式中的dilation。	INT64	ND
padding	属性	卷积的填充大小，对应公式中的padding。	INT64	ND
stride	属性	卷积的步长，对应公式中的stride。	INT64	ND
out	输出	梯度计算结果，对应公式中的input。数据类型和数据格式需要与`gradOutput`的数据类型和数据格式一致。	BFLOAT16、FLOAT16、FLOAT	5HD

约束说明

无

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_im2col_backward	通过aclnnIm2colBackward接口方式调用Col2im算子。