Skip to content

Commit 23dea76

Browse files
jinyouzhiCopilot
andauthored
【Hackathon 10th Spring No.1】add param for MaxPool*D/max_pool*d (#7724)
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
1 parent 65670ed commit 23dea76

22 files changed

Lines changed: 75 additions & 57 deletions

docs/api/paddle/nn/AdaptiveAvgPool2D_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ AdaptiveAvgPool2D
2525
2626
参数
2727
:::::::::
28-
- **output_size** (int|list|tuple):算子输出特征图的尺寸,如果其是 list 或 turple 类型的数值,必须包含两个元素,H 和 W。H 和 W 既可以是 int 类型值也可以是 None,None 表示与输入特征尺寸相同。
28+
- **output_size** (int|list|tuple):算子输出特征图的尺寸,如果其是 list 或 tuple 类型的数值,必须包含两个元素,H 和 W。H 和 W 既可以是 int 类型值也可以是 None,None 表示与输入特征尺寸相同。
2929
- **data_format** (str,可选):输入和输出的数据格式,可以是"NCHW"和"NHWC"。N 是批尺寸,C 是通道数,H 是特征高度,W 是特征宽度。默认值:"NCHW"。
3030
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
3131

docs/api/paddle/nn/AdaptiveAvgPool3D_cn.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ AdaptiveAvgPool3D
2828
2929
参数
3030
:::::::::
31-
- **output_size** (int|list|tuple):算子输出特征图的尺寸,如果其是 list 或 turple 类型的数值,必须包含三个元素,D,H 和 W。D,H 和 W 既可以是 int 类型值也可以是 None,None 表示与输入特征尺寸相同。
31+
- **output_size** (int|list|tuple):算子输出特征图的尺寸,如果其是 list 或 tuple 类型的数值,必须包含三个元素,D,H 和 W。D,H 和 W 既可以是 int 类型值也可以是 None,None 表示与输入特征尺寸相同。
3232
- **data_format** (str,可选):输入和输出的数据格式,可以是"NCDHW"和"NDHWC"。N 是批尺寸,C 是通道数,D 是特征长度,H 是特征高度,W 是特征宽度。默认值:"NCDHW"。
3333
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
3434

docs/api/paddle/nn/AdaptiveMaxPool1D_cn.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,12 @@ AdaptiveMaxPool1D
2222
参数
2323
:::::::::
2424
- **output_size** (int|list|tuple):算子输出特征图的长度,其数据类型为 int,list 或 tuple。
25-
- **return_mask** (bool,可选):如果设置为 True,则会与输出一起返回最大值的索引,默认为 False。
25+
- **return_mask** (bool,可选):如果设置为 True,则会与输出一起返回最大值的索引,默认为 False。别名 ``return_indices``。
2626
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
2727

2828
形状
2929
:::::::::
30-
- **x** (Tensor):默认形状为(批大小,通道数,输出特征长度),即 NCL 格式的 3-D Tensor。其数据类型为 float32 或者 float64。
30+
- **x** (Tensor):默认形状为(批大小,通道数,输出特征长度),即 NCL 格式的 3-D Tensor。其数据类型为 float32 或者 float64。别名 ``input``。
3131
- **output** (Tensor):默认形状为(批大小,通道数,输出特征长度),即 NCL 格式的 3-D Tensor。其数据类型与输入 x 相同。
3232

3333
返回

docs/api/paddle/nn/AdaptiveMaxPool2D_cn.rst

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,17 +27,18 @@ AdaptiveMaxPool2D
2727
2828
Output(i ,j) &= max(Input[hstart:hend, wstart:wend])
2929
30+
3031
参数
3132
:::::::::
3233

3334
- **output_size** (int|list|tuple):算子输出特征图的高和宽大小,其数据类型为 int,list 或 tuple。
34-
- **return_mask** (bool,可选):如果设置为 True,则会与输出一起返回最大值的索引,默认为 False。
35+
- **return_mask** (bool,可选):如果设置为 True,则会与输出一起返回最大值的索引,默认为 False。别名 ``return_indices``。
3536
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
3637

3738
形状
3839
:::::::::
3940

40-
- **x** (Tensor):默认形状为(批大小,通道数,输出特征长度,宽度),即 NCHW 格式的 4-D Tensor。其数据类型为 float32 或者 float64。
41+
- **x** (Tensor):默认形状为(批大小,通道数,输出特征长度,宽度),即 NCHW 格式的 4-D Tensor。其数据类型为 float32 或者 float64。别名 ``input``。
4142
- **output** (Tensor):默认形状为(批大小,通道数,输出特征长度,宽度),即 NCHW 格式的 4-D Tensor。其数据类型与输入 x 相同。
4243

4344
返回

docs/api/paddle/nn/AdaptiveMaxPool3D_cn.rst

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,15 +25,16 @@ AdaptiveMaxPool3D
2525
2626
Output(i ,j, k) &= max(Input[dstart:dend, hstart:hend, wstart:wend])
2727
28+
2829
参数
2930
:::::::::
3031
- **output_size** (int|list|tuple):算子输出特征图的高宽长大小,其数据类型为 int,list 或 tuple。
31-
- **return_mask** (bool,可选):如果设置为 True,则会与输出一起返回最大值的索引,默认为 False。
32+
- **return_mask** (bool,可选):如果设置为 True,则会与输出一起返回最大值的索引,默认为 False。别名 ``return_indices``。
3233
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
3334

3435
形状
3536
:::::::::
36-
- **x** (Tensor):默认形状为(批大小,通道数,输出特征深度,高度,宽度),即 NCDHW 格式的 5-D Tensor。其数据类型为 float32 或者 float64。
37+
- **x** (Tensor):默认形状为(批大小,通道数,输出特征深度,高度,宽度),即 NCDHW 格式的 5-D Tensor。其数据类型为 float32 或者 float64。别名 ``input``。
3738
- **output** (Tensor):默认形状为(批大小,通道数,输出特征深度,高度,宽度),即 NCDHW 格式的 5-D Tensor。其数据类型与输入 x 相同。
3839

3940
返回

docs/api/paddle/nn/FractionalMaxPool2D_cn.rst

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,19 +34,24 @@ FractionalMaxPool2D
3434
由于池化操作的步长与核尺寸相同,同为此随机序列,最终得到池化输出为 `[2, 4, 1, 5, 3]` 。
3535

3636

37+
.. note::
38+
别名支持: 参数名 ``input`` 可替代 ``x``,参数名 ``return_indices`` 可替代 ``return_mask``。
39+
3740
参数
3841
:::::::::
3942

4043
- **output_size** (int|list|tuple):算子输出图的尺寸,其数据类型为 int 或 list,tuple。如果输出为 tuple 或者 list,则必须包含两个元素, `(H, W)` 。 `H` 和 `W` 可以是 `int` ,也可以是 `None` ,表示与输入保持一致。
4144
- **kernel_size** (int|list|tuple, 可选) - 池化核大小。如果它是一个元组或列表,它必须包含两个整数值,(pool_size_Height, pool_size_Width)。若为一个整数,则表示 H 和 W 维度上均为该值,比如若 pool_size=2,则池化核大小为 [2,2]。默认为 `None`,表示使用 `disjoint` (`non-overlapping`) 模式。
4245
- **random_u** (float):分数阶池化操作的浮点随机数,取值范围为 `(0, 1)` 。默认为 `None` ,由框架随机生成,可以使用 `paddle.seed` 设置随机种子。
4346
- **return_mask** (bool,可选):如果设置为 `True` ,则会与输出一起返回最大值的索引,默认为 `False`。
47+
``别名:return_indices``
4448
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 `None`。
4549

4650
形状
4751
:::::::::
4852

4953
- **x** (Tensor):默认形状为(批大小,通道数,输出特征长度,宽度),即 NCHW 格式的 4-D Tensor。其数据类型为 float16, bfloat16, float32, float64。
54+
``别名:input``
5055
- **output** (Tensor):默认形状为(批大小,通道数,输出特征长度,宽度),即 NCHW 格式的 4-D Tensor。其数据类型与输入 x 相同。
5156

5257
返回

docs/api/paddle/nn/FractionalMaxPool3D_cn.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,12 +40,12 @@ FractionalMaxPool3D
4040
- **output_size** (int|list|tuple):算子输出图的尺寸,其数据类型为 int 或 list,tuple。如果输出为 tuple 或者 list,则必须包含两个元素, `(H, W)` 。 `H` 和 `W` 可以是 `int` ,也可以是 `None` ,表示与输入保持一致。
4141
- **kernel_size** (int|list|tuple,可选) - 池化核大小。如果它是一个元组或列表,它必须包含三个整数值,(pool_size_Depth,pool_size_Height, pool_size_Width)。若为一个整数,则表示 D,H 和 W 维度上均为该值,比如若 pool_size=2,则池化核大小为[2,2,2]。默认为 `None`,表示使用 `disjoint` (`non-overlapping`) 模式。
4242
- **random_u** (float):分数阶池化操作的浮点随机数,取值范围为 `(0, 1)` 。默认为 `None` ,由框架随机生成,可以使用 `paddle.seed` 设置随机种子。
43-
- **return_mask** (bool,可选):如果设置为 `True` ,则会与输出一起返回最大值的索引,默认为 `False`。
43+
- **return_mask** (bool,可选):如果设置为 `True` ,则会与输出一起返回最大值的索引,默认为 `False`。别名 ``return_indices``。
4444
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 `None`。
4545

4646
形状
4747
:::::::::
48-
- **x** (Tensor):默认形状为(批大小,通道数,输出特征深度,高度,宽度),即 NCDHW 格式的 5-D Tensor。其数据类型为 float16, bfloat16, float32, float64。
48+
- **x** (Tensor):默认形状为(批大小,通道数,输出特征深度,高度,宽度),即 NCDHW 格式的 5-D Tensor。其数据类型为 float16, bfloat16, float32, float64。别名 ``input``。
4949
- **output** (Tensor):默认形状为(批大小,通道数,输出特征深度,高度,宽度),即 NCDHW 格式的 5-D Tensor。其数据类型与输入 x 相同。
5050

5151
返回

docs/api/paddle/nn/MaxPool1D_cn.rst

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
MaxPool1D
44
-------------------------------
55

6-
.. py:class:: paddle.nn.MaxPool1D(kernel_size, stride=None, padding=0, return_mask=False, ceil_mode=False, name=None)
6+
.. py:class:: paddle.nn.MaxPool1D(kernel_size, stride=None, padding=0, return_mask=False, ceil_mode=False, dilation=1, name=None)
77
88
根据输入 `x` , `kernel_size` 等参数对一个输入 Tensor 计算 1D 的最大值池化。输入和输出都是 3-D Tensor,默认是以 `NCL` 格式表示的,其中 `N` 是 batch size,`C` 是通道数,`L` 是输入特征的长度。
99

@@ -13,19 +13,22 @@ MaxPool1D
1313
1414
Output(N_i, C_i, l) = max(Input[N_i, C_i, stride \times l:stride \times l+k])
1515
16+
1617
参数
1718
:::::::::
1819
- **kernel_size** (int|list|tuple) - 池化核的尺寸大小。如果 kernel_size 为 list 或 tuple 类型,其必须包含一个整数,最终池化核的大小为该数值。
1920
- **stride** (int|list|tuple,可选) - 池化操作步长。如果 stride 为 list 或 tuple 类型,其必须包含一个整数,最终池化操作的步长为该数值。默认值为 None,这时会使用 kernel_size 作为 stride。
2021
- **padding** (str|int|list|tuple,可选) - 池化补零的方式。如果 padding 是一个字符串,则必须为 `SAME` 或者 `VALID`。如果是 tuple 或者 list 类型,若其长度为 1,则会在输入的两端都填充上长度为 padding[0] 的 0,若其长度为 2,则应是 `[pad_before, pad_after]` 形式。如果 padding 是一个非 0 整数,那么表示会在输入的两端都填充上同样长度的 0。如果 padding 是长度为 2 的序列组成的 list 或 tuple,则其中的每个元素表示在相应维度上的填充长度。注意:在 batch_size 维度和通道维度应该设置为 [0, 0] 或 (0, 0)。默认值为 0。
21-
- **return_mask** (bool,可选) - 是否返回最大值的索引,默认为 False。
22+
- **return_mask** (bool,可选) - 是否返回最大值的索引,默认为 False。别名 ``return_indices``。
23+
``别名:return_indices``
2224
- **ceil_mode** (bool,可选) - 是否用 ceil 函数计算输出的 height 和 width,如果设置为 False,则使用 floor 函数来计算,默认为 False。
25+
- **dilation** (int|list|tuple,可选) - 空洞池化步长,如果 dilation 为 list 或 tuple 类型,其必须包含一个整数。默认值为 1。
2326
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
2427

2528

2629
形状
2730
:::::::::
28-
- **x** (Tensor) :默认形状为(批大小,通道数,长度),即 NCL 格式的 3-D Tensor。其数据类型为 float32 或 float64。
31+
- **input** (Tensor) :默认形状为(批大小,通道数,长度),即 NCL 格式的 3-D Tensor。其数据类型为 float32 或 float64。
2932
- **output** (Tensor) :默认形状为(批大小,通道数,输出特征长度),即 NCL 格式的 3-D Tensor。其数据类型与输入 x 相同。
3033

3134
返回

docs/api/paddle/nn/MaxPool2D_cn.rst

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
MaxPool2D
44
-------------------------------
55

6-
.. py:class:: paddle.nn.MaxPool2D(kernel_size, stride=None, padding=0, ceil_mode=False, return_mask=False, data_format="NCHW", name=None)
6+
.. py:class:: paddle.nn.MaxPool2D(kernel_size, stride=None, padding=0, return_mask=False, ceil_mode=False, dilation=1, data_format="NCHW", name=None)
77
构建 `MaxPool2D` 类的一个可调用对象,其将构建一个二维最大池化层,根据输入参数 `kernel_size`, `stride`,
88
`padding` 等参数对输入做最大池化操作。
99

@@ -27,16 +27,17 @@ MaxPool2D
2727
- **kernel_size** (int|list|tuple) - 池化核大小。如果它是一个元组或列表,它必须包含两个整数值,(pool_size_Height, pool_size_Width)。若为一个整数,则它的平方值将作为池化核大小,比如若 pool_size=2,则池化核大小为 2x2。
2828
- **stride** (int|list|tuple,可选) - 池化层的步长。如果它是一个元组或列表,它将包含两个整数,(pool_stride_Height, pool_stride_Width)。若为一个整数,则表示 H 和 W 维度上 stride 均为该值。默认值为 None,这时会使用 kernel_size 作为 stride。
2929
- **padding** (str|int|list|tuple,可选) - 池化填充。如果它是一个字符串,可以是"VALID"或者"SAME",表示填充算法。如果它是一个元组或列表,它可以有 3 种格式:(1)包含 2 个整数值:[pad_height, pad_width];(2)包含 4 个整数值:[pad_height_top, pad_height_bottom, pad_width_left, pad_width_right];(3)包含 4 个二元组:当 data_format 为"NCHW"时为 [[0,0], [0,0], [pad_height_top, pad_height_bottom], [pad_width_left, pad_width_right]],当 data_format 为"NHWC"时为[[0,0], [pad_height_top, pad_height_bottom], [pad_width_left, pad_width_right], [0,0]]。若为一个整数,则表示 H 和 W 维度上均为该值。默认值:0。
30+
- **return_mask** (bool,可选) - 是否返回最大索引和输出。默认为 False。别名 ``return_indices``。
3031
- **ceil_mode** (bool,可选) - 是否用 ceil 函数计算输出高度和宽度。如果是 True,则使用 `ceil` 计算输出形状的大小。
31-
- **return_mask** (bool,可选) - 是否返回最大索引和输出。默认为 False
32+
- **dilation** (int|list|tuple,可选) - 空洞池化步长,如果 dilation 为 list 或 tuple 类型,其必须包含两个整数:[dilation_height, dilation_width]。若为一个整数,则表示 H 和 W 维度上均为该值。默认值:1
3233
- **data_format** (str,可选) - 输入和输出的数据格式,可以是"NCHW"和"NHWC"。N 是批尺寸,C 是通道数,H 是特征高度,W 是特征宽度。默认值:"NCHW"。
3334
- **name** (str,可选) - 具体用法请参见 :ref:`api_guide_Name`,一般无需设置,默认值为 None。
3435

3536

3637

3738
形状
3839
:::::::::
39-
- **x** (Tensor) :默认形状为(批大小,通道数,高度,宽度),即 NCHW 格式的 4-D Tensor。其数据类型为 float16, float32, float64, int32 或 int64。
40+
- **x** (Tensor) :默认形状为(批大小,通道数,高度,宽度),即 NCHW 格式的 4-D Tensor。其数据类型为 float16, float32, float64, int32 或 int64。别名 ``input``。
4041
- **output** (Tensor) :默认形状为(批大小,通道数,输出特征高度,输出特征宽度),即 NCHW 格式的 4-D Tensor。其数据类型与输入相同。
4142

4243

0 commit comments

Comments
 (0)