adaptive_pool2d

    该OP使用上述输入参数的池化配置,为二维空间自适应池化操作,根据 input , 池化类型 pool_type , 池化核大小 pool_size 这些参数得到输出。

    输入X和输出Out是NCHW格式,N为批大小,C是通道数,H是特征高度,W是特征宽度。参数 含有两个整型元素, 分别代表高度和宽度上的参数。输出Out的H和W维由 pool_size 决定,即输出shape为

    对于平均adaptive pool2d:

    adaptive_pool2d - 图1

    • input (Variable) - 池化操作的输入张量,维度为

      的4-D Tensor。 输入张量的格式为NCHW,其中N是batch大小,C是通道数,H是特征的高度,W是特征的宽度,数据类型为float32或float64。

    • pool_size (int|list|tuple) - 池化核大小。 如果池化核大小是元组或列表,则它必须包含两个整数(pool_size_Height,pool_size_Width。若为一个整数,则表示H和W维度上均为该值。

    • name (str,可选) – 具体用法请参见 ,一般无需设置。默认值:None。

    返回: Variable(Tensor) 自适应池化结果张量

    返回类型:变量(Variable),数据类型与 input 一致

    抛出异常: