Function max_pool2d

imperative/python/megengine/functional/nn.py:669–712

r"""Applies a 2D max pooling over an input tensor. Refer to :class:`~.MaxPool2d` for more information. Args: inp: input tensor of shape :math:`(N, C, H_{\text{in}}, W_{\text{in}})`. kernel_size: size of the window used to calculate the max value. stride: stride of t

(
    inp: Tensor,
    kernel_size: Union[int, Tuple[int, int]],
    stride: Optional[Union[int, Tuple[int, int]]] = None,
    padding: Union[int, Tuple[int, int]] = 0,
)

def max_pool2d(
    inp: Tensor,
    kernel_size: Union[int, Tuple[int, int]],
    stride: Optional[Union[int, Tuple[int, int]]] = None,
    padding: Union[int, Tuple[int, int]] = 0,
) -> Tensor:
    r"""Applies a 2D max pooling over an input tensor.

    Refer to :class:`~.MaxPool2d` for more information.

    Args:
        inp: input tensor of shape :math:`(N, C, H_{\text{in}}, W_{\text{in}})`.
        kernel_size: size of the window used to calculate the max value.
        stride: stride of the window. Default value is ``kernel_size``.
        padding: implicit zero padding added on both sides. Default: 0.

    Returns:
        output tensor of shape `(N, C, H_{\text{out}}, W_{\text{out}})`.

    Examples:
        >>> import numpy as np
        >>> input = Tensor(np.arange(1 * 1 * 3 * 4).astype(np.float32).reshape(1, 1, 3, 4))
        >>> F.nn.max_pool2d(input, 2, 1, 0)
        Tensor([[[[ 5.  6.  7.]
           [ 9. 10. 11.]]]], device=xpux:0)
    """
    if stride is None:
        stride = kernel_size
    window_h, window_w = expand_hw(kernel_size)
    stride_h, stride_w = expand_hw(stride)
    padding_h, padding_w = expand_hw(padding)

    op = builtin.Pooling(
        window_h=window_h,
        window_w=window_w,
        stride_h=stride_h,
        stride_w=stride_w,
        pad_h=padding_h,
        pad_w=padding_w,
        mode="max",
        strategy=get_execution_strategy(),
    )
    (output,) = apply(op, inp)
    return output


def avg_pool2d(