提交 94de2290 编写于 作者: J jerrywgz

fix format in api doc, test=develop

上级 0e1b426c
...@@ -6904,13 +6904,13 @@ def prelu(x, mode, param_attr=None, name=None): ...@@ -6904,13 +6904,13 @@ def prelu(x, mode, param_attr=None, name=None):
Args: Args:
x (Variable): The input tensor. x (Variable): The input tensor.
param_attr(ParamAttr|None): The parameter attribute for the learnable param_attr(ParamAttr|None): The parameter attribute for the learnable
weight (alpha). weight (alpha).
mode (string): The mode for weight sharing. It supports all, channel mode (string): The mode for weight sharing. It supports all, channel
and element. all: all elements share same weight and element. all: all elements share same weight
channel:elements in a channel share same weight channel:elements in a channel share same weight
element:each element has a weight element:each element has a weight
name(str|None): A name for this layer(optional). If set None, the layer name(str|None): A name for this layer(optional). If set None, the layer
will be named automatically. will be named automatically.
Returns: Returns:
...@@ -6920,9 +6920,9 @@ def prelu(x, mode, param_attr=None, name=None): ...@@ -6920,9 +6920,9 @@ def prelu(x, mode, param_attr=None, name=None):
.. code-block:: python .. code-block:: python
x = fluid.layers.data(name="x", shape=[10,10], dtype="float32") x = fluid.layers.data(name="x", shape=[10,10], dtype="float32")
mode = 'channel' mode = 'channel'
output = fluid.layers.prelu(x,mode) output = fluid.layers.prelu(x,mode)
""" """
helper = LayerHelper('prelu', **locals()) helper = LayerHelper('prelu', **locals())
if mode not in ['all', 'channel', 'element']: if mode not in ['all', 'channel', 'element']:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册