From 162cf75c88111b44e1126a84b8b6438f5dac752c Mon Sep 17 00:00:00 2001 From: barrierye Date: Mon, 5 Nov 2018 10:48:39 +0800 Subject: [PATCH] Submit PR again test=develop --- .../paddle/fluid/tests/unittests/test_similarity_focus_op.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_similarity_focus_op.py b/python/paddle/fluid/tests/unittests/test_similarity_focus_op.py index bd3b2782aea..b3833f05f1a 100755 --- a/python/paddle/fluid/tests/unittests/test_similarity_focus_op.py +++ b/python/paddle/fluid/tests/unittests/test_similarity_focus_op.py @@ -57,8 +57,7 @@ class TestSimilarityFocusOp(OpTest): if cnt == min(y_dim, z_dim): break channel[index] = -1 - res = res.reshape(1, y_dim, z_dim) - res = res.repeat([x_dim], axis=0) + res = res.reshape(1, y_dim, z_dim).repeat([x_dim], axis=0) res = res.reshape(1, x_dim, y_dim, z_dim) if output is not None: output = np.concatenate((output, res), axis=0) -- GitLab