prelu_cn.rst 1.4 KB
Newer Older
H
Hao Wang 已提交
1 2 3 4 5
.. _cn_api_fluid_layers_prelu:

prelu
-------------------------------

6
:api_attr: 声明式编程(静态图)专用API
7

H
Hao Wang 已提交
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
.. py:function:: paddle.fluid.layers.prelu(x, mode, param_attr=None, name=None)

等式:

.. math::
    y = max(0, x) + \alpha min(0, x)

共提供三种激活方式:

.. code-block:: text

    all: 所有元素使用同一个alpha值
    channel: 在同一个通道中的元素使用同一个alpha值
    element: 每一个元素有一个独立的alpha值


参数:
25 26 27
          - **x** (Variable)- 多维Tensor或LoDTensor,数据类型为float32。
          - **mode** (str) - 权重共享模式。
          - **param_attr** (ParamAttr,可选) - 可学习权重 :math:`[\alpha]` 的参数属性,可由ParamAttr创建。默认值为None,表示使用默认的权重参数属性。具体用法请参见 :ref:`cn_api_fluid_ParamAttr` 。
W
wangguanzhong 已提交
28
          - **name** (str,可选) – 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None。 
H
Hao Wang 已提交
29 30


31 32 33 34 35
返回: 表示激活输出Tensor或LoDTensor,数据类型为float32。与输入形状相同。


返回类型:Variable

H
Hao Wang 已提交
36 37 38 39 40 41 42

**代码示例:**

.. code-block:: python

    import paddle.fluid as fluid
    from paddle.fluid.param_attr import ParamAttr
W
wangguanzhong 已提交
43
    x = fluid.data(name="x", shape=[None,5,10,10], dtype="float32")
H
Hao Wang 已提交
44 45 46 47 48
    mode = 'channel'
    output = fluid.layers.prelu(
             x,mode,param_attr=ParamAttr(name='alpha'))