diff --git a/doc/api/v2/config/layer.rst b/doc/api/v2/config/layer.rst index 46f81bf9a2d92154e1f90201ac0730fcb5a378d3..65d4c472c35f0fe726c32e4745b7352b2fc87d4d 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 8044c7aa92a94f130863c8e349762e2d0d7799ac..bb58135084df5dfc76aec1e6d790e29ab0c10a88 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(