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.]