未验证 提交 ed52b009 编写于 作者: D Double_V 提交者: GitHub

fix avg_pool3d count_include_pad as True (#27189)

上级 d488181d
...@@ -165,7 +165,6 @@ class TestPool3d_API(unittest.TestCase): ...@@ -165,7 +165,6 @@ class TestPool3d_API(unittest.TestCase):
self.assertTrue(np.allclose(result.numpy(), result_np)) self.assertTrue(np.allclose(result.numpy(), result_np))
def check_max_dygraph_ndhwc_results(self, place): def check_max_dygraph_ndhwc_results(self, place):
print("run ndchw max pool3d")
with fluid.dygraph.guard(place): with fluid.dygraph.guard(place):
input_np = np.random.random([2, 3, 32, 32, 32]).astype("float32") input_np = np.random.random([2, 3, 32, 32, 32]).astype("float32")
input = fluid.dygraph.to_variable( input = fluid.dygraph.to_variable(
...@@ -190,7 +189,6 @@ class TestPool3d_API(unittest.TestCase): ...@@ -190,7 +189,6 @@ class TestPool3d_API(unittest.TestCase):
np.transpose(result.numpy(), [0, 4, 1, 2, 3]), result_np)) np.transpose(result.numpy(), [0, 4, 1, 2, 3]), result_np))
def check_max_dygraph_ceilmode_results(self, place): def check_max_dygraph_ceilmode_results(self, place):
print("run ceil mode max pool3d")
with fluid.dygraph.guard(place): with fluid.dygraph.guard(place):
input_np = np.random.random([2, 3, 32, 32, 32]).astype("float32") input_np = np.random.random([2, 3, 32, 32, 32]).astype("float32")
input = fluid.dygraph.to_variable(input_np) input = fluid.dygraph.to_variable(input_np)
......
...@@ -389,7 +389,7 @@ def avg_pool3d(x, ...@@ -389,7 +389,7 @@ def avg_pool3d(x,
stride=None, stride=None,
padding=0, padding=0,
ceil_mode=False, ceil_mode=False,
count_include_pad=False, count_include_pad=True,
divisor_override=None, divisor_override=None,
data_format="NCDHW", data_format="NCDHW",
name=None): name=None):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册