平均池化层
【输入】
X:一个张量,数据类型:float16、float32,格式为NCHW
【输出】
Y:一个张量,数据类型:float16、float32,格式为NCHW
【属性】
auto_pad:可选,支持NOTSET、SAME_UPPER、SAME_LOWER与VALID
count_include_pad:int,暂不支持
kernel_shape:可选,包括:
− kernel_shape[0]:数据类型:int32,指定沿H维度的窗口大小,取值范围为[1, 32768],默认为1
− kernel_shape[1]:数据类型:int32,指定沿W维度的窗口大小,取值范围为[1, 32768],默认为1
strides:可选,包括:
− strides[0]:数据类型:int32,指定沿H维度的步长,默认为1
− strides[1]:数据类型:int32,指定沿W维度的步长,默认为1
pads:可选,包括:
− pads[0]:数据类型:int32,指定顶部padding,默认为0
− pads[1]:数据类型:int32,指定底部padding,默认为0
− pads[2]:数据类型:int32,指定左部padding,默认为0
− pads[3]:数据类型:int32,指定右部padding,默认为0
ceil_mode:可选,数据类型:int32,取值:0(floor模式),1(ceil模式),默认为0
【约束】
strides[0]或者strides[1]取值步长大于63时,会使用AI CPU计算,性能会下降;
kernel_shape_H或kernel_shape_W取值超过[1,255],或者kernel_shape_H * kernel_shape_W > 256时,会使用AI CPU计算,导致性能下降;
1 <= input_w <= 4096;
当输入张量的N是一个质数时,N应当小于65535;
ceil_mode参数仅在auto_pad='NOTSET'时生效;
不支持atc工具参数--precision_mode=must_keep_origin_dtype时float32类型输入;
auto_pad属性值SAME_UPPER, SAME_LOWER统一使用的TBE的SAME属性,即TBE算子没有根据这个属性区分pad的填充位置,可能会带来精度问题
Opset v8/v9/v10/v11/v12/v13