prelu

注意:该API仅支持【静态图】模式

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

等式:

prelu - 图1

共提供三种激活方式:

  1. all: 所有元素使用同一个alpha
  2. channel: 在同一个通道中的元素使用同一个alpha
  3. element: 每一个元素有一个独立的alpha
  • 参数:
    • x (Variable)- 多维Tensor或LoDTensor,数据类型为float32。
    • mode (str) - 权重共享模式。
    • param_attr (ParamAttr,可选) - 可学习权重 prelu - 图2 的参数属性,可由ParamAttr创建。默认值为None,表示使用默认的权重参数属性。具体用法请参见 ParamAttr
    • name (str,可选) – 具体用法请参见 Name ,一般无需设置,默认值为None。

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

返回类型:Variable

代码示例:

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