2022-02-19 14:42:47 +08:00
|
|
|
|
mindspore.nn.ReLU6
|
|
|
|
|
|
===================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:class:: mindspore.nn.ReLU6
|
|
|
|
|
|
|
|
|
|
|
|
ReLU6激活函数。
|
|
|
|
|
|
|
|
|
|
|
|
ReLU6类似于ReLU，不同之处在于设置了上限，其上限为6，如果输入大于6，输出会被限制为6。公式如下：
|
|
|
|
|
|
|
|
|
|
|
|
.. math::
|
2022-03-28 19:08:15 +08:00
|
|
|
|
Y = \min(\max(0, x), 6).
|
2022-02-19 14:42:47 +08:00
|
|
|
|
|
|
|
|
|
|
**输入：**
|
|
|
|
|
|
|
2022-03-28 19:08:15 +08:00
|
|
|
|
- **x** (Tensor) - ReLU6的输入，任意维度的Tensor，其数据类型为float16或float32。
|
2022-02-19 14:42:47 +08:00
|
|
|
|
|
|
|
|
|
|
**输出：**
|
|
|
|
|
|
|
|
|
|
|
|
Tensor，数据类型与 `x` 相同。
|
|
|
|
|
|
|
|
|
|
|
|
**异常：**
|
|
|
|
|
|
|
|
|
|
|
|
- **TypeError** - `x` 的数据类型既不是float16也不是float32。
|
