From 3757c1ee479ddb67b622db03b41d3300f3fab62d Mon Sep 17 00:00:00 2001 From: shippingwang Date: Tue, 25 Dec 2018 17:10:54 +0000 Subject: [PATCH] Modify test layers, test=develop --- python/paddle/fluid/layers/nn.py | 7 +++---- python/paddle/fluid/tests/unittests/test_layers.py | 4 ++-- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py index 7f7086641ac..63ff5f07e7b 100644 --- a/python/paddle/fluid/layers/nn.py +++ b/python/paddle/fluid/layers/nn.py @@ -9341,8 +9341,8 @@ def shuffle_channel(x, group=1, name=None): with multiple group convolutional layers. Args: - x: The input tensor variable.. - group: The num of group + x(Variable): The input tensor variable. + group(Integer): The num of group. Returns: Variable: channels shuffled tensor variable. @@ -9358,8 +9358,7 @@ def shuffle_channel(x, group=1, name=None): """ helper = LayerHelper("shuffle_channel", **locals()) - out = helper.create_variable_for_type_inference( - dtype=helper.input_dtype('X')) + out = helper.create_variable_for_type_inference(dtype=x.dtype) if not isinstance(group, int): raise TypeError("group must be int type") diff --git a/python/paddle/fluid/tests/unittests/test_layers.py b/python/paddle/fluid/tests/unittests/test_layers.py index daf4a9c824a..9e392fa8e0d 100644 --- a/python/paddle/fluid/tests/unittests/test_layers.py +++ b/python/paddle/fluid/tests/unittests/test_layers.py @@ -1018,8 +1018,8 @@ class TestBook(unittest.TestCase): def test_shuffle_channel(self): program = Program() with program_guard(program): - x = layers.data(name="X", shape=[10, 16, 4, 4], dtype="float32") - out = layers.shuffle_channel(x, group=2) + x = layers.data(name="X", shape=[16, 4, 4], dtype="float32") + out = layers.shuffle_channel(x, group=4) self.assertIsNotNone(out) print(str(program)) -- GitLab