# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. import paddle from ...fluid.data_feeder import convert_dtype, check_variable_and_dtype, check_type, check_dtype from ...fluid.layers import utils from ...fluid.dygraph import layers from ...fluid.layer_helper import LayerHelper from .. import functional as F __all__ = [ 'AdaptiveAvgPool2d', 'AdaptiveAvgPool3d', 'AvgPool1d', 'maxPool1d', 'AdaptiveMaxPool1d', 'AdaptiveAvgPool1d', 'AvgPool2d', 'MaxPool2d', 'AvgPool3d', 'MaxPool3d', ] class AdaptiveAvgPool2d(layers.Layer): """ This operation applies 2D adaptive avg pooling on input tensor. The h and w dimensions of the output tensor are determined by the parameter output_size. For avg adaptive pool2d: .. math:: hstart &= floor(i * H_{in} / H_{out}) hend &= ceil((i + 1) * H_{in} / H_{out}) wstart &= floor(j * W_{in} / W_{out}) wend &= ceil((j + 1) * W_{in} / W_{out}) Output(i ,j) &= \\frac{sum(Input[hstart:hend, wstart:wend])}{(hend - hstart) * (wend - wstart)} Parameters: output_size (int|list|tuple): The pool kernel size. If pool kernel size is a tuple or list, it must contain two element, (H, W). H and W can be either a int, or None which means the size will be the same as that of the input. data_format (str): The data format of the input and output data. An optional string from: "NCHW", "NHWC". The default is "NCHW". When it is "NCHW", the data is stored in the order of: [batch_size, input_channels, input_height, input_width]. name(str, optional): For detailed information, please refer to :ref:`api_guide_Name`. Usually name is no need to set and None by default. Shape: x (Tensor): The input tensor of adaptive avg pool2d operator, which is a 4-D tensor. The data type can be float32 or float64. output (Tensor): The output tensor of adaptive avg pool2d operator, which is a 4-D tensor. The data type is same as input x. Returns: A callable object of AdaptiveAvgPool2d. Examples: .. code-block:: python # adaptive avg pool2d # suppose input data in shape of [N, C, H, W], `output_size` is [m, n], # output shape is [N, C, m, n], adaptive pool divide H and W dimensions # of input data into m * n grids averagely and performs poolings in each # grid to get output. # adaptive avg pool performs calculations as follow: # # for i in range(m): # for j in range(n): # hstart = floor(i * H / m) # hend = ceil((i + 1) * H / m) # wstart = floor(i * W / n) # wend = ceil((i + 1) * W / n) # output[:, :, i, j] = avg(input[:, :, hstart: hend, wstart: wend]) # import paddle import numpy as np paddle.disable_static() input_data = np.random.rand(2, 3, 32, 32) x = paddle.to_tensor(input_data) # x.shape is [2, 3, 32, 32] adaptive_avg_pool = paddle.nn.AdaptiveAvgPool2d(output_size=3) pool_out = adaptive_avg_pool(x = x) # pool_out.shape is [2, 3, 3, 3] """ def __init__(self, output_size, data_format="NCHW", name=None): super(AdaptiveAvgPool2d, self).__init__() self._output_size = output_size self._data_format = data_format self._name = name def forward(self, x): return F.adaptive_avg_pool2d( x, output_size=self._output_size, data_format=self._data_format, name=self._name) class AdaptiveAvgPool3d(layers.Layer): """ This operation applies 3D adaptive avg pooling on input tensor. The h and w dimensions of the output tensor are determined by the parameter output_size. For avg adaptive pool3d: .. math:: dstart &= floor(i * D_{in} / D_{out}) dend &= ceil((i + 1) * D_{in} / D_{out}) hstart &= floor(j * H_{in} / H_{out}) hend &= ceil((j + 1) * H_{in} / H_{out}) wstart &= floor(k * W_{in} / W_{out}) wend &= ceil((k + 1) * W_{in} / W_{out}) Output(i ,j, k) &= \\frac{sum(Input[dstart:dend, hstart:hend, wstart:wend])}{(dend - dstart) * (hend - hstart) * (wend - wstart)} Parameters: output_size (int|list|tuple): The pool kernel size. If pool kernel size is a tuple or list, it must contain three elements, (D, H, W). D, H and W can be either a int, or None which means the size will be the same as that of the input. data_format (str): The data format of the input and output data. An optional string from: "NCDHW", "NDHWC". The default is "NCDHW". When it is "NCDHW", the data is stored in the order of: [batch_size, input_channels, input_depth, input_height, input_width]. name(str, optional): For detailed information, please refer to :ref:`api_guide_Name`. Usually name is no need to set and None by default. Shape: x (Tensor): The input tensor of adaptive avg pool3d operator, which is a 5-D tensor. The data type can be float32 or float64. output (Tensor): The output tensor of adaptive avg pool3d operator, which is a 5-D tensor. The data type is same as input x. Returns: A callable object of AdaptiveAvgPool3d. Examples: .. code-block:: python # adaptive avg pool3d # suppose input data in shape of [N, C, D, H, W], `output_size` is [l, m, n], # output shape is [N, C, l, m, n], adaptive pool divide D, H and W dimensions # of input data into l * m * n grids averagely and performs poolings in each # grid to get output. # adaptive avg pool performs calculations as follow: # # for i in range(l): # for j in range(m): # for k in range(n): # dstart = floor(i * D / l) # dend = ceil((i + 1) * D / l) # hstart = floor(j * H / m) # hend = ceil((j + 1) * H / m) # wstart = floor(k * W / n) # wend = ceil((k + 1) * W / n) # output[:, :, i, j, k] = # avg(input[:, :, dstart:dend, hstart: hend, wstart: wend]) import paddle import numpy as np paddle.disable_static() input_data = np.random.rand(2, 3, 8, 32, 32) x = paddle.to_tensor(input_data) # x.shape is [2, 3, 8, 32, 32] adaptive_avg_pool = paddle.nn.AdaptiveAvgPool3d(output_size=3) pool_out = adaptive_avg_pool(x = x) # pool_out = [2, 3, 3, 3, 3] """ def __init__(self, output_size, data_format="NCDHW", name=None): super(AdaptiveAvgPool3d, self).__init__() self._output_size = output_size self._data_format = data_format self._name = name def forward(self, x): return F.adaptive_avg_pool3d( x, output_size=self._output_size, data_format=self._data_format, name=self._name) class AvgPool1d(layers.Layer): """ This operation applies a 1D average pooling over an input signal composed of several input planes, based on the input, output_size, return_indices parameters. Input(X) and output(Out) are in NCL format, where N is batch size, C is the number of channels, L is the length of the feature. The output tensor shape will be [N, C, output_size]. The output value of the layer with input size (N, C, L), output (N, C, L_{out}) and kernel_size k can be precisely described as For average pool1d: .. math:: Output(N_i, C_i, l) &= mean(Input[N_i, C_i, stride \times l:stride \times l+k]) Args: kernel_size (int|list|tuple): The pool kernel size. If pool kernel size is a tuple or list, it must contain one integers. stride (int|list|tuple): The pool stride size. If pool stride size is a tuple or list, it must contain one integers. padding (string|int|list|tuple): The pool padding. If `pool_padding` is a string, either 'VALID' or 'SAME' which is the padding algorithm. If pool padding size is a tuple or list, it could be the following forms: `[pad_left, pad_right]`. If padding is non-zero, then the input is implicitly zero-padded on both sides for padding number of points. count_include_pad (bool): Whether to exclude padding points in average pooling mode, default is `true`. ceil_mode (bool): ${ceil_mode_comment}Whether to use the ceil function to calculate output height and width. If it is set to False, the floor function will be used. Default False name(str, optional): For detailed information, please refer to :ref:`api_guide_Name`. Usually name is no need to set and None by default. Returns: None. Raises: ValueError: If `padding` is a string, but not "SAME" or "VALID". ValueError: If `padding` is "VALID", but `ceil_mode` is True. ValueError: If `padding` is a list or tuple but its length greater than 1. ShapeError: If the input is not a 3-D. ShapeError: If the output's shape calculated is not greater than 0. Examples: .. code-block:: python import paddle import paddle.nn as nn paddle.disable_static() data = paddle.to_tensor(np.random.uniform(-1, 1, [1, 3, 32]).astype(np.float32)) AvgPool1d = nn.AvgPool1d(kernel_size=2, stride=2, padding=0) pool_out = AvgPool1d(data) # pool_out shape: [1, 3, 16] """ def __init__(self, kernel_size, stride=None, padding=0, count_include_pad=True, ceil_mode=False, name=None): super(AvgPool1d, self).__init__() self.kernel_size = kernel_size self.stride = stride self.padding = padding self.ceil_mode = ceil_mode self.count_include_pad = count_include_pad self.name = name def forward(self, x): out = F.avg_pool1d(x, self.kernel_size, self.stride, self.padding, self.count_include_pad, self.ceil_mode, self.name) return out class MaxPool1d(layers.Layer): """ Applies a 1D max pooling over an input signal composed of several input planes based on the input, output_size, return_indices parameters. Input(X) and output(Out) are in NCL format, where N is batch size, C is the number of channels, L is the length of the feature. The output value of the layer with input size (N, C, L), output (N, C, L_{out}) and kernel_size k can be precisely described as For average pool1d: .. math:: Output(N_i, C_i, l) &= max(Input[N_i, C_i, stride \times l:stride \times l+k])} Args: kernel_size (int|list|tuple): The pool kernel size. If pool kernel size is a tuple or list, it must contain one integers. stride (int|list|tuple): The pool stride size. If pool stride size is a tuple or list, it must contain one integers. padding (string|int|list|tuple): The pool padding. If `pool_padding` is a string, either 'VALID' or 'SAME' which is the padding algorithm. If pool padding size is a tuple or list, it could be the following forms: `[pad_left, pad_right]`. return_indices (bool): Whether return the max indices along with the outputs. default is `False`. ceil_mode (bool): Whether to use the ceil function to calculate output height and width. False is the default. If it is set to False, the floor function will be used. Default False name(str, optional): For detailed information, please refer to :ref:`api_guide_Name`. Usually name is no need to set and None by default. Returns: None. Raises: ValueError: If `padding` is a string, but not "SAME" or "VALID". ValueError: If `padding` is "VALID", but `ceil_mode` is True. ValueError: If `padding` is a list or tuple but its length greater than 1. ShapeError: If the input is not a 3-D. ShapeError: If the output's shape calculated is not greater than 0. Examples: .. code-block:: python import paddle import paddle.nn as nn paddle.disable_static() data = paddle.to_tensor(np.random.uniform(-1, 1, [1, 3, 32]).astype(np.float32)) MaxPool1d = nn.MaxPool1d(kernel_size=2, stride=2, padding=0) pool_out = MaxPool1d(data) # pool_out shape: [1, 3, 16] MaxPool1d = nn.MaxPool1d(kernel_size=2, stride=2, padding=0, return_indices=True) pool_out, indices = MaxPool1d(data) # pool_out shape: [1, 3, 16], indices shape: [1, 3, 16] """ def __init__(self, kernel_size, stride=None, padding=0, return_indices=False, ceil_mode=False, name=None): super(MaxPool1d, self).__init__() self.kernel_size = kernel_size self.stride = stride self.padding = padding self.ceil_mode = ceil_mode self.return_indices = return_indices self.name = name def forward(self, input): out = F.max_pool1d(input, self.kernel_size, self.stride, self.padding, self.return_indices, self.ceil_mode, self.name) return out class AdaptiveAvgPool1d(layers.Layer): """ This operation applies a 1D adaptive average pooling over an input signal composed of several input planes, based on the input, output_size, return_indices parameters. Input(X) and output(Out) are in NCL format, where N is batch size, C is the number of channels, L is the length of the feature. The output tensor shape will be [N, C, output_size]. For average adaptive pool1d: .. math:: lstart &= floor(i * L_{in} / L_{out}) lend &= ceil((i + 1) * L_{in} / L_{out}) Output(i) &= \\frac{sum(Input[lstart:lend])}{(lstart - lend)} Args: output_size (int|list|tuple): The pool kernel size. If pool kernel size is a tuple or list, it must contain one int. name(str, optional): For detailed information, please refer to :ref:`api_guide_Name`. Usually name is no need to set and None by default. Returns: None. Raises: ValueError: 'pool_size' should be a integer or list or tuple with length as 1. Examples: .. code-block:: python # average adaptive pool1d # suppose input data in shape of [N, C, L], `output_size` is m or [m], # output shape is [N, C, m], adaptive pool divide L dimension # of input data into m grids averagely and performs poolings in each # grid to get output. # adaptive max pool performs calculations as follow: # # for i in range(m): # lstart = floor(i * L / m) # lend = ceil((i + 1) * L / m) # output[:, :, i] = sum(input[:, :, lstart: lend])/(lstart - lend) # import paddle import paddle.nn as nn paddle.disable_static() data = paddle.to_tensor(np.random.uniform(-1, 1, [1, 3, 32]).astype(np.float32)) AdaptiveAvgPool1d = nn.AdaptiveAvgPool1d(output_size=16) pool_out = AdaptiveAvgPool1d(data) # pool_out shape: [1, 3, 16] """ def __init__(self, output_size, name=None): super(AdaptiveAvgPool1d, self).__init__() self.output_size = output_size self.name = name def forward(self, input): return F.adaptive_avg_pool1d(input, self.output_size, self.name) class AdaptiveMaxPool1d(layers.Layer): """ This operation applies a 1D adaptive max pooling over an input signal composed of several input planes, based on the input, output_size, return_indices parameters. Input(X) and output(Out) are in NCL format, where N is batch size, C is the number of channels, L is the length of the feature. The output tensor shape will be [N, C, output_size]. For max adaptive pool1d: .. math:: lstart &= floor(i * L_{in} / L_{out}) lend &= ceil((i + 1) * L_{in} / L_{out}) Output(i) &= max(Input[lstart:lend])} Args: output_size (int|list|tuple): The pool kernel size. If pool kernel size is a tuple or list, it must contain one int. return_indices (bool): If true, the index of max pooling point will be returned along with outputs. It cannot be set in average pooling type. Default False. name(str, optional): For detailed information, please refer to :ref:`api_guide_Name`. Usually name is no need to set and None by default. Returns: None. Raises: ValueError: 'pool_size' should be a integer or list or tuple with length as 1. Examples: .. code-block:: python # max adaptive pool1d # suppose input data in shape of [N, C, L], `output_size` is m or [m], # output shape is [N, C, m], adaptive pool divide L dimension # of input data into m grids averagely and performs poolings in each # grid to get output. # adaptive max pool performs calculations as follow: # # for i in range(m): # lstart = floor(i * L / m) # lend = ceil((i + 1) * L / m) # output[:, :, i] = max(input[:, :, lstart: lend]) # import paddle import paddle.nn as nn paddle.disable_static() data = paddle.to_tensor(np.random.uniform(-1, 1, [1, 3, 32]).astype(np.float32)) AdaptiveMaxPool1d = nn.AdaptiveMaxPool1d(output_size=16) pool_out = AdaptiveMaxPool1d(data) # pool_out shape: [1, 3, 16] # for return_indices = true AdaptiveMaxPool1d = nn.AdaptiveMaxPool1d(output_size=16, return_indices=True) pool_out, indices = AdaptiveMaxPool1d(data) # pool_out shape: [1, 3, 16], indices shape: [1, 3, 16] """ def __init__(self, output_size, return_indices=False, name=None): super(AdaptiveMaxPool1d, self).__init__() self.output_size = output_size self.return_indices = return_indices self.name = name def forward(self, input): return F.adaptive_max_pool1d(input, self.output_size, self.return_indices, self.name) class AvgPool2d(layers.Layer): """ This operation applies 2D average pooling over input features based on the input, and kernel_size, stride, padding parameters. Input(X) and Output(Out) are in NCHW format, where N is batch size, C is the number of channels, H is the height of the feature, and W is the width of the feature. Example: Input: X shape: $(N, C, H_{in}, W_{in})$ Attr: kernel_size: ksize Output: Out shape: $(N, C, H_{out}, W_{out})$ $$ out(N_i, C_j, h, w) = \frac{1}{ksize[0] * ksize[1]} \sum_{m=0}^{ksize[0]-1} \sum_{n=0}^{ksize[1]-1} input(N_i, C_j, stride[0] \times h + m, stride[1] \times w + n) $$ Args: kernel_size (int|list|tuple): The pool kernel size. If pool kernel size is a tuple or list, it must contain two integers, (pool_size_Height, pool_size_Width). Otherwise, the pool kernel size will be a square of an int. stride (int|list|tuple): The pool stride size. If pool stride size is a tuple or list, it must contain two integers, (pool_stride_Height, pool_stride_Width). Otherwise, the pool stride size will be a square of an int. Default: kernel_size. padding (string|int|list|tuple): The pool padding. If `pool_padding` is a string, either 'VALID' or 'SAME' which is the padding algorithm. If pool padding size is a tuple or list, it could be in three forms: `[pad_height, pad_width]` or `[pad_height_top, pad_height_bottom, pad_width_left, pad_width_right]`, and when `data_format` is `"NCHW"`, `pool_padding` can be in the form `[[0,0], [0,0], [pad_height_top, pad_height_bottom], [pad_width_left, pad_width_right]]`. when `data_format` is `"NHWC"`, `pool_padding` can be in the form `[[0,0], [pad_height_top, pad_height_bottom], [pad_width_left, pad_width_right], [0,0]]`. Otherwise, the pool padding size will be a square of an int. ceil_mode (bool): when True, will use `ceil` instead of `floor` to compute the output shape count_include_pad (bool): Whether to exclude padding points in average pooling mode, default is `true`. divisor_override (int|float) if specified, it will be used as divisor, otherwise kernel_size will be used. Default None. name(str, optional): For detailed information, please refer to :ref:`api_guide_Name`. Usually name is no need to set and None by default. data_format (string): The data format of the input and output data. An optional string from: `"NCHW"`, `"NDHW"`. The default is `"NCHW"`. When it is `"NCHW"`, the data is stored in the order of: `[batch_size, input_channels, input_height, input_width]`. Returns: None. Raises: ValueError: If `padding` is a string, but not "SAME" or "VALID". ValueError: If `padding` is "VALID", but `ceil_mode` is True. ShapeError: If the output's shape calculated is not greater than 0. Examples: .. code-block:: python import paddle import paddle.nn as nn import numpy as np paddle.disable_static() # max pool2d input = paddle.to_tensor(np.random.uniform(-1, 1, [1, 3, 32, 32]).astype(np.float32)) AvgPool2d = nn.AvgPool2d(kernel_size=2, stride=2, padding=0) output = AvgPoo2d(input) # output.shape [1, 3, 16, 16] """ def __init__(self, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True, divisor_override=None, data_format="NCHW", name=None): super(AvgPool2d, self).__init__() self.ksize = kernel_size self.stride = stride self.padding = padding self.ceil_mode = ceil_mode self.count_include_pad = count_include_pad self.divisor = divisor_override self.data_format = data_format self.name = name def forward(self, x): return F.avg_pool2d( x, kernel_size=self.ksize, stride=self.stride, padding=self.padding, ceil_mode=self.ceil_mode, count_include_pad=self.count_include_pad, divisor_override=self.divisor, data_format=self.data_format, name=self.name) class MaxPool2d(layers.Layer): """ This operation applies 2D max pooling over input feature based on the input, and kernel_size, stride, padding parameters. Input(X) and Output(Out) are in NCHW format, where N is batch size, C is the number of channels, H is the height of the feature, and W is the width of the feature. Example: Input: X shape: $(N, C, H_{in}, W_{in})$ Attr: kernel_size: ksize Output: Out shape: $(N, C, H_{out}, W_{out})$ $$ out(N_i, C_j, h, w) ={} & \max_{m=0, \ldots, ksize[0] -1} \max_{n=0, \ldots, ksize[1]-1} \\ & \text{input}(N_i, C_j, \text{stride[0]} \times h + m, \text{stride[1]} \times w + n) $$ Args: kernel_size (int|list|tuple): The pool kernel size. If pool kernel size is a tuple or list, it must contain two integers, (pool_size_Height, pool_size_Width). Otherwise, the pool kernel size will be a square of an int. stride (int|list|tuple): The pool stride size. If pool stride size is a tuple or list, it must contain two integers, (pool_stride_Height, pool_stride_Width). Otherwise, the pool stride size will be a square of an int. Default: kernel_size. padding (string|int|list|tuple): The pool padding. If `pool_padding` is a string, either 'VALID' or 'SAME' which is the padding algorithm. If pool padding size is a tuple or list, it could be in three forms: `[pad_height, pad_width]` or `[pad_height_top, pad_height_bottom, pad_width_left, pad_width_right]`, and when `data_format` is `"NCHW"`, `pool_padding` can be in the form `[[0,0], [0,0], [pad_height_top, pad_height_bottom], [pad_width_left, pad_width_right]]`. when `data_format` is `"NHWC"`, `pool_padding` can be in the form `[[0,0], [pad_height_top, pad_height_bottom], [pad_width_left, pad_width_right], [0,0]]`. Otherwise, the pool padding size will be a square of an int. ceil_mode (bool): when True, will use `ceil` instead of `floor` to compute the output shape return_indices (bool): Whether to return the max indices along with the outputs. data_format (string): The data format of the input and output data. An optional string from: `"NCHW"`, `"NDHW"`. The default is `"NCHW"`. When it is `"NCHW"`, the data is stored in the order of: `[batch_size, input_channels, input_height, input_width]`. name(str, optional): For detailed information, please refer to :ref:`api_guide_Name`. Usually name is no need to set and None by default. Returns: None Raises: ValueError: If `padding` is a string, but not "SAME" or "VALID". ValueError: If `padding` is "VALID", but `ceil_mode` is True. ShapeError: If the output's shape calculated is not greater than 0. Examples: .. code-block:: python import paddle import paddle.nn as nn import numpy as np paddle.disable_static() # max pool2d input = paddle.to_tensor(np.random.uniform(-1, 1, [1, 3, 32, 32]).astype(np.float32)) MaxPool2d = nn.MaxPool2d(kernel_size=2, stride=2, padding=0) output = MaxPool2d(input) # output.shape [1, 3, 16, 16] # for return_indices=True MaxPool2d = nn.MaxPool2d(kernel_size=2,stride=2, padding=0, return_indices=True) output, max_indices = MaxPool2d(input) # output.shape [1, 3, 16, 16], max_indices.shape [1, 3, 16, 16], """ def __init__(self, kernel_size, stride=None, padding=0, return_indices=False, ceil_mode=False, data_format="NCHW", name=None): super(MaxPool2d, self).__init__() self.ksize = kernel_size self.stride = stride self.padding = padding self.return_indices = return_indices self.ceil_mode = ceil_mode self.data_format = data_format self.name = name def forward(self, x): return F.max_pool2d( x, kernel_size=self.ksize, stride=self.stride, padding=self.padding, return_indices=self.return_indices, data_format=self.data_format, name=self.name) class MaxPool3d(layers.Layer): """ This operation applies 3D max pooling over input features based on the input, and kernel_size, stride, padding parameters. Input(X) and Output(Out) are in NCDHW format, where N is batch size, C is the number of channels, H is the height of the feature, D is the depth of the feature, and W is the width of the feature. Args: kernel_size (int|list|tuple): The pool kernel size. If pool kernel size is a tuple or list, it must contain three integers, (pool_size_Depth, pool_size_Height, pool_size_Width). Otherwise, the pool kernel size will be the cube of an int. stride (string|int|list|tuple)): The pool padding. If `pool_padding` is a string, either 'VALID' or 'SAME' which is the padding algorithm. If pool stride size is a tuple or list, it must contain three integers, `[stride_Depth, stride_Height, stride_Width]`. Otherwise, the pool stride size will be a cube of an int. Default kernel_size. padding (int|list|tuple): The pool padding size. If pool padding size is a tuple or list, it could be in three forms: `[pad_depth, pad_height, pad_width]` or `[pad_depth_front, pad_depth_back, pad_height_top, pad_height_bottom, pad_width_left, pad_width_right]`, and when `data_format` is `"NCDHW"`, `pool_padding` can be in the form `[[0,0], [0,0], [pad_depth_front, pad_depth_back], [pad_height_top, pad_height_bottom], [pad_width_left, pad_width_right]]`. when `data_format` is `"NDHWC"`, `pool_padding` can be in the form `[[0,0], [pad_depth_front, pad_depth_back], [pad_height_top, pad_height_bottom], [pad_width_left, pad_width_right], [0,0]]`. ceil_mode (bool): when True, will use ceil instead of floor to compute the output shape. count_include_pad (bool): Whether to exclude padding points in average pooling mode, default is True. data_format (string): The data format of the input and output data. An optional string from: `"NCHW"`, `"NDHW"`. The default is `"NCHW"`. When it is `"NCHW"`, the data is stored in the order of: `[batch_size, input_channels, input_height, input_width]`. name(str, optional): For detailed information, please refer to :ref:`api_guide_Name`. Usually name is no need to set and None by default. Returns:None. Raises: ValueError: If `padding` is a string, but not "SAME" or "VALID". ValueError: If `padding` is "VALID", but `ceil_mode` is True. ShapeError: If the output's shape calculated is not greater than 0. Examples: .. code-block:: python import paddle import paddle.nn as nn import numpy as np paddle.disable_static() # max pool3d input = paddle.to_tensor(np.random.uniform(-1, 1, [1, 2, 3, 32, 32]).astype(np.float32)) MaxPool3d = nn.MaxPool3d(kernel_size=2, stride=2, padding=0) output = MaxPool3d(input) # output.shape [1, 2, 3, 16, 16] # for return_indices=True MaxPool3d = nn.MaxPool3d(kernel_size=2,stride=2, padding=0, return_indices=True) output, max_indices = MaxPool3d(input) # output.shape [1, 2, 3, 16, 16], max_indices.shape [1, 2, 3, 16, 16], """ def __init__(self, kernel_size, stride, padding, return_indices=False, ceil_mode=False, data_format="NCDHW", name=None): super(MaxPool3d, self).__init__() self.ksize = kernel_size self.stride = stride self.padding = padding self.return_indices = return_indices self.ceil_mode = ceil_mode self.data_format = data_format self.name = name def forward(self, x): return F.max_pool3d( x, kernel_size=self.ksize, stride=self.stride, padding=self.padding, return_indices=self.return_indices, data_format=self.data_format, name=self.name) class AvgPool3d(layers.Layer): """ This operation applies 3D max pooling over input features based on the input, and kernel_size, stride, padding parameters. Input(X) and Output(Out) are in NCDHW format, where N is batch size, C is the number of channels, H is the height of the feature, D is the depth of the feature, and W is the width of the feature. Args: kernel_size (int|list|tuple): The pool kernel size. If pool kernel size is a tuple or list, it must contain three integers, (pool_size_Depth, pool_size_Height, pool_size_Width). Otherwise, the pool kernel size will be the cube of an int. stride (string|int|list|tuple)): The pool padding. If `pool_padding` is a string, either 'VALID' or 'SAME' which is the padding algorithm. If pool stride size is a tuple or list, it must contain three integers, `[stride_Depth, stride_Height, stride_Width]`. Otherwise, the pool stride size will be a cube of an int. padding (int|list|tuple): The pool padding size. If pool padding size is a tuple or list, it could be in three forms: `[pad_depth, pad_height, pad_width]` or `[pad_depth_front, pad_depth_back, pad_height_top, pad_height_bottom, pad_width_left, pad_width_right]`, and when `data_format` is `"NCDHW"`, `pool_padding` can be in the form `[[0,0], [0,0], [pad_depth_front, pad_depth_back], [pad_height_top, pad_height_bottom], [pad_width_left, pad_width_right]]`. when `data_format` is `"NDHWC"`, `pool_padding` can be in the form `[[0,0], [pad_depth_front, pad_depth_back], [pad_height_top, pad_height_bottom], [pad_width_left, pad_width_right], [0,0]]`. ceil_mode (bool): ${ceil_mode_comment} count_include_pad (bool): Whether to exclude padding points in average pooling mode, default is True. divisor_override (int|float) if specified, it will be used as divisor, otherwise kernel_size will be used. Default None. data_format (string): The data format of the input and output data. An optional string from: `"NCHW"`, `"NDHW"`. The default is `"NCHW"`. When it is `"NCHW"`, the data is stored in the order of: `[batch_size, input_channels, input_height, input_width]`. name(str, optional): For detailed information, please refer to :ref:`api_guide_Name`. Usually name is no need to set and None by default. Returns: None. Raises: ValueError: If `padding` is a string, but not "SAME" or "VALID". ValueError: If `padding` is "VALID", but `ceil_mode` is True. ShapeError: If the output's shape calculated is not greater than 0. Examples: .. code-block:: python import paddle import paddle.nn as nn import numpy as np paddle.disable_static() # avg pool3d input = paddle.to_tensor(np.random.uniform(-1, 1, [1, 2, 3, 32, 32]).astype(np.float32)) AvgPool3d = nn.AvgPool3d(kernel_size=2, stride=2, padding=0) output = AvgPool3d(input) # output.shape [1, 2, 3, 16, 16] """ def __init__(self, kernel_size, stride, padding=0, ceil_mode=False, count_include_pad=True, divisor_override=None, data_format="NCDHW", name=None): super(AvgPool3d, self).__init__() self.ksize = kernel_size self.stride = stride self.padding = padding self.ceil_mode = ceil_mode self.count_include_pad = count_include_pad self.divisor = divisor_override self.data_format = data_format self.name = name def forward(self, x): return F.avg_pool3d( x, kernel_size=self.ksize, stride=self.stride, padding=self.padding, ceil_mode=self.ceil_mode, count_include_pad=self.count_include_pad, divisor_override=self.divisor, data_format=self.data_format, name=self.name)