提交 3e9ea348 编写于 作者: Z Zhaolong Xing 提交者: Yi Wang

fix prelu doc (#5807)

上级 f7fc6c2d
...@@ -6640,8 +6640,10 @@ def prelu_layer(input, ...@@ -6640,8 +6640,10 @@ def prelu_layer(input,
:type partial_sum: int :type partial_sum: int
:param channel_shared: whether or not the parameter are shared across channels. :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 = 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. - channel_shared = False, we set the partial_sum to the number of elements in one channel.
:type channel_shared: bool :type channel_shared: bool
:param num_channels: number of input channel. :param num_channels: number of input channel.
:type num_channels: int :type num_channels: int
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册