relu6

paddle.fluid.layers.relu6(x, threshold=6.0, name=None)[源代码]

relu6激活函数

relu6 - 图1

参数

  • x (Variable) - 输入的多维 Tensor ,数据类型为:float32、float64。
  • threshold (float) - relu6的阈值。默认值为6.0
  • name (str,可选) – 具体用法请参见 Name ,一般无需设置,默认值为None。

返回

x 维度相同、数据类型相同的 Tensor

返回类型

Variable

代码示例

  1. import paddle.fluid as fluid
  2. import numpy as np
  3. in1 = np.array([[-1,0],[2.5,7.8]])
  4. with fluid.dygraph.guard():
  5. x1 = fluid.dygraph.to_variable(in1)
  6. out1 = fluid.layers.relu6(x=x1, threshold=6.0)
  7. print(out1.numpy())
  8. # [[0. 0. ]
  9. # [2.5 6. ]]