From 0e60bb3c4ffdca714b73196c1a4eb5385b9b87a7 Mon Sep 17 00:00:00 2001 From: barrierye Date: Wed, 31 Oct 2018 10:54:27 +0800 Subject: [PATCH] Submit PR again test=develop --- .../paddle/fluid/tests/unittests/test_similarity_focus_op.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 b3833f05f1a..bd3b2782aea 100755 --- a/python/paddle/fluid/tests/unittests/test_similarity_focus_op.py +++ b/python/paddle/fluid/tests/unittests/test_similarity_focus_op.py @@ -57,7 +57,8 @@ class TestSimilarityFocusOp(OpTest): if cnt == min(y_dim, z_dim): break channel[index] = -1 - res = res.reshape(1, y_dim, z_dim).repeat([x_dim], axis=0) + res = res.reshape(1, y_dim, z_dim) + res = res.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