PaddlePaddle

API 指南

 paddle.nn / ReLU6

ReLU6

class paddle.nn. ReLU6 ( name=None ) [源代码]

ReLU6激活层

\[ReLU6(x) = min(max(0,x), 6)\]

其中，\(x\) 为输入的 Tensor

参数

  • name (str, 可选) - 操作的名称(可选，默认值为None）。更多信息请参见 Name。

形状:

  • input: 任意形状的Tensor。

  • output: 和input具有相同形状的Tensor。

代码示例

import paddle
import numpy as np

x = paddle.to_tensor(np.array([-1, 0.3, 6.5]))
m = paddle.nn.ReLU6()
out = m(x) # [0, 0.3, 6]

此页内容是否对您有帮助

上一篇 下一篇
©Copyright 2011-2021 www.yanxishe.com ©AI研习社 All Rights Reserved 粤ICP备11095991号-21