prelu¶

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

paddle.fluid.layers.prelu(x, mode, param_attr=None, name=None)[源代码]¶

等式：

\[y = max(0, x) + \alpha min(0, x)\]

共提供三种激活方式：

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

参数¶

x （Variable）- 多维Tensor或LoDTensor，数据类型为float32。

mode (str) - 权重共享模式。

param_attr (ParamAttr，可选) - 可学习权重 \([\alpha]\) 的参数属性，可由ParamAttr创建。默认值为None，表示使用默认的权重参数属性。具体用法请参见 ParamAttr 。

name (str，可选) – 具体用法请参见 Name ，一般无需设置，默认值为None。

返回¶

表示激活输出Tensor或LoDTensor，数据类型为float32。与输入形状相同。

返回类型¶

Variable

代码示例¶

import paddle.fluid as fluid
from paddle.fluid.param_attr import ParamAttr
x = fluid.data(name="x", shape=[None,5,10,10], dtype="float32")
mode = 'channel'
output = fluid.layers.prelu(
         x,mode,param_attr=ParamAttr(name='alpha'))