平均池化层。
一个输入:
x:一个tensor,数据类型:float16、float,格式为NCHW。
一个输出:
y:一个tensor,数据类型:float16、float,格式为NCHW。
auto_pad:string,可选,支持NOTSET、SAME_UPPER与VALID。
count_include_pad:int,暂不支持。
kernel_shape:可选,包括:
strides:可选,包括:
pads:可选,包括:
ceil_mode:可选,数据类型:int32,取值:0(floor模式),1(ceil模式),默认为0。
strides[0]或者strides[1]取值步长大于63时,会使用AI CPU计算,性能会下降。
kernel_shape[0]或kernel_shape[1]取值超过[1,255],或者kernel_shape[0] * kernel_shape[1] > 256时,会使用AI CPU计算,导致性能下降。
1 <= input_w <= 4096。
当输入tensor的N是一个质数时,N应当小于65535。
ceil_mode参数仅在auto_pad='NOTSET'时生效。
不支持atc工具参数--precision_mode=must_keep_origin_dtype时float类型输入。
auto_pad属性值支持SAME_UPPER而不支持SAME_LOWER,原因是SAME_LOWER和SAME_UPPER统一使用的TBE的SAME属性,即TBE算子没有根据这个属性区分pad的填充位置,可能会带来精度问题。
滑窗范围超过输入特征图原始宽高的大小,且count_include_pad为False时,可能导致算子计算中的除数分母为0,输出结果可能为0,65504,Nan,INF。该场景不符合该算子的正常业务逻辑,建议修改ceil_mode或stride等属性,满足滑窗始终与输入特征图的有交集。
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18