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-08-26 11:32:26 +08:00
|
|
|
|
input是具有任何有效形状的张量。
|
|
|
|
|
|
|
2022-07-25 16:47:23 +08:00
|
|
|
|
输入:
|
|
|
|
|
|
- **x** (Tensor) - ReLU6的输入,任意维度的Tensor,其数据类型为float16或float32。
|
2022-06-28 16:46:24 +08:00
|
|
|
|
|
2022-07-25 16:47:23 +08:00
|
|
|
|
输出:
|
|
|
|
|
|
Tensor,数据类型与 `x` 相同。
|
2022-02-19 14:42:47 +08:00
|
|
|
|
|
2022-07-25 16:47:23 +08:00
|
|
|
|
异常:
|
|
|
|
|
|
- **TypeError** - `x` 的数据类型既不是float16也不是float32。
|