Dilation2D

产品支持情况

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

功能说明

• 算子功能：形态学膨胀（Morphological Dilation）操作的二维实现。对输入图像的每个通道独立执行滑窗操作：在每个窗口位置，取输入值与滤波器值之和的最大值作为输出。

• 计算公式：

$$y[b,oh,ow,c]=\underset{fh,fw}{\max }(x[b,oh\times strid{e}_h+fh\times rat{e}_h,ow\times strid{e}_w+fw\times rat{e}_w,c]+filter[fh,fw,c])$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	4D输入张量，NHWC或NCHW格式。NHWC时filter shape为(fH,fW,C)，NCHW时为(C,fH,fW)。	FLOAT16	ND
filter	输入	3D滤波器张量，C维度与x的C一致。NHWC时shape为(fH,fW,C)，NCHW时为(C,fH,fW)。	与x相同	ND
y	输出	4D输出张量，oH/oW由输入shape+属性推导。	与x相同	ND
strides	属性	4个正整数，滑窗步幅[sN, sH, sW, sC]。	ListInt	-
rates	属性	4个正整数，膨胀率[rN, rH, rW, rC]。	ListInt	-
padding_mode	属性	padding模式，"SAME"或"VALID"，默认"SAME"。	String	-
pads	属性	4个非负整数，显式padding值，默认{0,0,0,0}。	ListInt	-
ceil_mode	属性	输出尺寸计算使用ceil还是floor，默认false。	Bool	-
data_format	属性	数据格式，"NHWC"或"NCHW"，默认"NHWC"。	String	-

约束说明

• 输入x必须为4D张量，filter必须为3D张量。
• x和filter的dtype必须一致，仅支持FLOAT16。
• filter的通道数（C维度）必须与x的通道数一致。NHWC时filter shape为(fH,fW,C)，NCHW时为(C,fH,fW)。
• strides和rates必须为4个正整数。NHWC时strides[0]=strides[3]=1, rates[0]=rates[3]=1；NCHW时strides[0]=strides[1]=1, rates[0]=rates[1]=1。
• CALCULATED模式下pads必须小于有效滤波器尺寸，pads格式为[H_top, H_bottom, W_left, W_right]。
• 超出输入边界的位置视为负无穷，不参与max比较。
• 支持SAME、VALID和CALCULATED三种padding模式。
• 整数类型加法采用饱和截断（saturate），防止溢出wrap-around。
• 仅支持Ascend 950。

调用说明

调用方式	样例代码	说明
图模式接口	test_geir_dilation2_d	通过算子IR构图方式调用Dilation2D算子。