Source code for webdnn.graph.operators.max_pooling_2d

from typing import Optional

from webdnn.graph.operators.pooling_2d import Pooling2D
from webdnn.graph.operators.util import IntOrTuple


class MaxPooling2D(Pooling2D):
    """MaxPooling2D(name, ksize, stride, padding, cover_all=False)

    Spatial max pooling operator.

    Args:
        name (str): Operator name.
        ksize (int or tuple of int): Kernel size.
        stride (int or tuple of int): Stride size.
        padding (int or tuple of int): Padding size.
        cover_all (bool, optional): If `True`, all input pixels are pooled into some output pixels.

    Signature

        .. code::

            y, = op(x)

        - **x** - Input variable.
        - **y** - Output value. Its order is same as :code:`x`.
    """

    def __init__(self, name: Optional[str], ksize: IntOrTuple, stride: IntOrTuple, padding: IntOrTuple, cover_all: bool = True):
        super(MaxPooling2D, self).__init__(name, ksize, stride, padding, cover_all)