Im2col

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

功能说明

• 算子功能：图像到列，滑动局部窗口数据转为列向量，拼接为大张量。从批处理输入张量中提取滑动窗口。考虑一个形状为（N, C, H, W）或 (C, H, W) 的批处理输入张量，其中N是批处理维度， C是通道维度， 而 H, W 表示图像大小，此操作将输入的空间维度内的每个滑动kernel_size大小的块展平为（N, C $\times \prod$（kernel_size）, L）的3-D 或 （C $\times \prod$（kernel_size）, L）的2-D 的 output张量的列（即最后一维），而L是这些块的总数。
• 计算公式： $L={\prod }_d\lfloor \frac{spatia{l}_{s}ize[d]+2\times padding[d]-dilation[d]\times （kerne{l}_{s}ize[d]-1）-1}{stride[d]}+1$ \rfloor, 其中spatial_size由上述输入张量的H,W构成。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
self	输入张量	输入张量，shape为3维或4维。	FLOAT、FLOAT16、BFLOAT16	ND
kernelSize	输入数组	卷积核的大小，size为2。	INT64	-
dilation	输入数组	膨胀参数，size为2。	INT64	-
padding	输入数组	填充大小，size为2，padding[0]表示H方向，padding[1]表示W方向。	INT64	-
stride	输入数组	步长，size为2。	INT64	-
out	输出张量	输出张量，shape根据参数推导得出。	FLOAT、FLOAT16、BFLOAT16	ND

约束说明

• 输入张量的维度必须是3维或4维。
• kernelSize、dilation、padding、stride的size必须为2。
• kernelSize、dilation、stride的值必须大于0。
• padding的值不能小于0。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_im2col	通过aclnnIm2col接口方式调用Im2col算子。