From 3e9ea348217c2fa95d53f56b678e7dc50d7ca845 Mon Sep 17 00:00:00 2001 From: Zhaolong Xing Date: Wed, 22 Nov 2017 09:47:23 +0800 Subject: [PATCH] fix prelu doc (#5807) --- python/paddle/trainer_config_helpers/layers.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/paddle/trainer_config_helpers/layers.py b/python/paddle/trainer_config_helpers/layers.py index 71407973025..6bd5ce4fe2f 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 -- GitLab