diff --git a/python/paddle/trainer_config_helpers/layers.py b/python/paddle/trainer_config_helpers/layers.py index 714079730250e7fd886838c377c8952d18a4348e..6bd5ce4fe2f70946befb388986dff603bdae0b8e 100644 --- a/python/paddle/trainer_config_helpers/layers.py +++ b/python/paddle/trainer_config_helpers/layers.py @@ -6640,8 +6640,10 @@ def prelu_layer(input, :type partial_sum: int :param channel_shared: whether or not the parameter are shared across channels. + - channel_shared = True, we set the partial_sum to the number of outputs. - channel_shared = False, we set the partial_sum to the number of elements in one channel. + :type channel_shared: bool :param num_channels: number of input channel. :type num_channels: int