diff --git a/python/paddle/trainer_config_helpers/layers.py b/python/paddle/trainer_config_helpers/layers.py index 7140797302..6bd5ce4fe2 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