Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
未验证 提交 17db031a 编写于 作者: D Double_V 提交者： GitHub
浏览文件
操作

fix pool bug, test=develop (#28359)

上级 6115c14f
显示空白变更内容
内联 并排
Showing with 5 addition and 3 deletion
python/paddle/nn/functional/pooling.py
浏览文件 @ 17db031a
...@@ -230,7 +230,7 @@ def avg_pool1d(x, ...@@ -230,7 +230,7 @@ def avg_pool1d(x,
x, 'pooling_type', 'avg', 'ksize', kernel_size, 'global_pooling', x, 'pooling_type', 'avg', 'ksize', kernel_size, 'global_pooling',
False, 'strides', stride, 'paddings', padding, 'padding_algorithm', False, 'strides', stride, 'paddings', padding, 'padding_algorithm',
padding_algorithm, 'use_cudnn', True, 'ceil_mode', ceil_mode, padding_algorithm, 'use_cudnn', True, 'ceil_mode', ceil_mode,
'use_mkldnn', False, 'exclusive', not exclusive, 'data_format', 'use_mkldnn', False, 'exclusive', exclusive, 'data_format',
data_format) data_format)
return squeeze(output, [2]) return squeeze(output, [2])
...@@ -338,7 +338,7 @@ def avg_pool2d(x, ...@@ -338,7 +338,7 @@ def avg_pool2d(x,
x, 'pooling_type', 'avg', 'ksize', kernel_size, 'global_pooling', x, 'pooling_type', 'avg', 'ksize', kernel_size, 'global_pooling',
False, 'padding_algorithm', padding_algorithm, 'strides', stride, False, 'padding_algorithm', padding_algorithm, 'strides', stride,
'paddings', padding, 'use_cudnn', True, 'ceil_mode', ceil_mode, 'paddings', padding, 'use_cudnn', True, 'ceil_mode', ceil_mode,
'use_mkldnn', False, 'exclusive', not exclusive, 'data_format', 'use_mkldnn', False, 'exclusive', exclusive, 'data_format',
data_format) data_format)
if divisor_override is None: if divisor_override is None:
return output return output
...@@ -452,7 +452,7 @@ def avg_pool3d(x, ...@@ -452,7 +452,7 @@ def avg_pool3d(x,
x, 'pooling_type', 'avg', 'ksize', kernel_size, 'strides', stride, x, 'pooling_type', 'avg', 'ksize', kernel_size, 'strides', stride,
'paddings', padding, 'global_pooling', False, 'padding_algorithm', 'paddings', padding, 'global_pooling', False, 'padding_algorithm',
padding_algorithm, 'use_cudnn', True, 'ceil_mode', ceil_mode, padding_algorithm, 'use_cudnn', True, 'ceil_mode', ceil_mode,
'use_mkldnn', False, 'exclusive', not exclusive, 'data_format', 'use_mkldnn', False, 'exclusive', exclusive, 'data_format',
data_format) data_format)
if divisor_override is None: if divisor_override is None:
return output return output
......
python/paddle/nn/layer/pooling.py
浏览文件 @ 17db031a
...@@ -503,6 +503,7 @@ class MaxPool2D(layers.Layer): ...@@ -503,6 +503,7 @@ class MaxPool2D(layers.Layer):
stride=self.stride, stride=self.stride,
padding=self.padding, padding=self.padding,
return_mask=self.return_mask, return_mask=self.return_mask,
ceil_mode=self.ceil_mode,
data_format=self.data_format, data_format=self.data_format,
name=self.name) name=self.name)
...@@ -594,6 +595,7 @@ class MaxPool3D(layers.Layer): ...@@ -594,6 +595,7 @@ class MaxPool3D(layers.Layer):
stride=self.stride, stride=self.stride,
padding=self.padding, padding=self.padding,
return_mask=self.return_mask, return_mask=self.return_mask,
ceil_mode=self.ceil_mode,
data_format=self.data_format, data_format=self.data_format,
name=self.name) name=self.name)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或