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

fix avg_pool3d count_include_pad as True (#27189)

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