mindspore.nn.ReLU

class mindspore.nn.ReLU[源代码]

修正线性单元激活函数（Rectified Linear Unit activation function）。

逐元素求 \(\max(x,\ 0)\) 。特别说明，负数输出值会被修改为0，正数输出不受影响。

\[\text{ReLU}(x) = (x)^+ = \max(0, x),\]

ReLU相关图参见 ReLU 。

输入：

  • x (Tensor) - 用于计算ReLU的任意维度的Tensor。数据类型为 number。

输出：

Tensor，数据类型和shape与 x 相同。

异常：

  • TypeError - x 的数据类型不是number。

支持平台：

Ascend GPU CPU

样例：

>>> x = Tensor(np.array([-1, 2, -3, 2, -1]), mindspore.float16)
>>> relu = nn.ReLU()
>>> output = relu(x)
>>> print(output)
[0. 2. 0. 2. 0.]