mindspore.nn.ReLU6

查看源文件
class mindspore.nn.ReLU6[源代码]

ReLU6激活函数。

ReLU6类似于ReLU,不同之处在于设置了上限,其上限为6,如果输入大于6,输出会被限制为6。公式如下:

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

ReLU6函数图:

../../_images/ReLU6.png
输入:
  • x (Tensor) - ReLU6的输入,是具有任何有效形状的张量,其数据类型为float16或float32。

输出:

Tensor,数据类型与 x 相同。

异常:
  • TypeError - x 的数据类型既不是float16也不是float32。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> from mindspore import Tensor, nn
>>> import numpy as np
>>> x = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float16)
>>> relu6 = nn.ReLU6()
>>> output = relu6(x)
>>> print(output)
[0. 0. 0. 2. 1.]