提交 1cbda104 编写于 作者: S songyouwei 提交者: zhongpu

update PRelu doc (#1687)

* update PRelu doc
test=develop

* refine doc
test=develop

* split arg
test=develop

* fix alpha format
test=develop
上级 fede1399
......@@ -16,13 +16,14 @@ PRelu
参数:
- **mode** (str) - 权重共享模式。共提供三种激活方式:
.. code-block:: text
all:所有元素使用同一个 :math:`[\alpha]` 值
channel:在同一个通道中的元素使用同一个 :math:`[\alpha]` 值
element:每一个元素有一个独立的 :math:`[\alpha]` 值
.. code-block:: text
- **input_shape** (list 或 tuple,可选) - 输入的维度,该参数仅在mode参数为"all"时需要设置。默认为None。
all:所有元素使用同一个alpha值
channel:在同一个通道中的元素使用同一个alpha值
element:每一个元素有一个独立的alpha值
- **channel** (int,可选) - 通道数。该参数在mode参数为"channel"时是必须的。默认为None。
- **input_shape** (int 或 list 或 tuple,可选) - 输入的维度。该参数在mode参数为"element"时是必须的。默认为None。
- **param_attr** (ParamAttr, 可选) - 指定权重参数属性的对象。默认值为None,表示使用默认的权重参数属性。具体用法请参见 :ref:`cn_api_fluid_ParamAttr` 。
- **dtype** (str, 可选) - 数据类型,可以为"float32"或"float64"。默认值:"float32"。
......@@ -39,9 +40,20 @@ PRelu
inp_np = np.ones([5, 200, 100, 100]).astype('float32')
with fluid.dygraph.guard():
inp_np = to_variable(inp_np)
mode = 'channel'
prelu = fluid.PRelu('prelu', mode=mode, param_attr=fluid.ParamAttr(initializer=fluid.initializer.Constant(1.0)))
dy_rlt = prelu(inp_np)
prelu0 = fluid.PRelu(
mode='all',
param_attr=fluid.ParamAttr(initializer=fluid.initializer.Constant(1.0)))
dy_rlt0 = prelu0(inp_np)
prelu1 = fluid.PRelu(
mode='channel',
channel=200,
param_attr=fluid.ParamAttr(initializer=fluid.initializer.Constant(1.0)))
dy_rlt1 = prelu1(inp_np)
prelu2 = fluid.PRelu(
mode='element',
input_shape=inp_np.shape,
param_attr=fluid.ParamAttr(initializer=fluid.initializer.Constant(1.0)))
dy_rlt2 = prelu2(inp_np)
属性
::::::::::::
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册