From 50764480c1a98cfe43d64874a0324c1b309146f7 Mon Sep 17 00:00:00 2001
From: caoying03 <caoying03@baidu.com>
Date: Thu, 8 Jun 2017 10:18:14 +0800
Subject: [PATCH] follow comments.

---
 doc/api/v2/config/layer.rst                    |  2 +-
 python/paddle/trainer_config_helpers/layers.py | 17 +++--------------
 2 files changed, 4 insertions(+), 15 deletions(-)

diff --git a/doc/api/v2/config/layer.rst b/doc/api/v2/config/layer.rst
index 46f81bf9a2..65d4c472c3 100644
--- a/doc/api/v2/config/layer.rst
+++ b/doc/api/v2/config/layer.rst
@@ -443,7 +443,7 @@ eos
     :noindex:
 
 Activation with learnable parameter
-===================
+===================================
 
 prelu
 --------
diff --git a/python/paddle/trainer_config_helpers/layers.py b/python/paddle/trainer_config_helpers/layers.py
index 8044c7aa92..bb58135084 100755
--- a/python/paddle/trainer_config_helpers/layers.py
+++ b/python/paddle/trainer_config_helpers/layers.py
@@ -5595,24 +5595,13 @@ def prelu_layer(input,
     :rtype: LayerOutput
     """
 
-    if isinstance(input, collections.Sequence):
-        assert len(input) == 1, 'prelu_layer only accepts one input'
-    else:
-        input = [input]
-    assert isinstance(input[0], LayerOutput)
-
-    if isinstance(param_attr, collections.Sequence):
-        assert len(param_attr) == 1, (
-            'because prelu_layer only accepts one input '
-            'it requires only one parameter setting.')
-    else:
-        param_attr = [param_attr]
-    assert isinstance(param_attr[0], ParameterAttribute)
+    assert isinstance(input, LayerOutput), 'prelu_layer only accepts one input'
+    assert isinstance(param_attr, ParameterAttribute)
 
     l = Layer(
         name=name,
         type='prelu',
-        inputs=Input(input[0].name, **param_attr[0].attr),
+        inputs=Input(input.name, **param_attr.attr),
         partial_sum=partial_sum,
         **ExtraLayerAttribute.to_kwargs(layer_attr))
     return LayerOutput(
-- 
GitLab