mindspore.ops.ReLUV2
- class mindspore.ops.ReLUV2[源代码]
线性修正单元激活函数(Rectified Linear Unit activation function)。
按元素返回 \(\max(x,\ 0)\) 。特别说明,负数输出值会被修改为0,正数输出不受影响。
\[\text{ReLU}(x) = (x)^+ = \max(0, x),\]输入:
input_x (Tensor) - 输入Tensor必须是4-D Tensor。
输出:
output (Tensor) - 数据类型和shape与 input_x 的相同。
mask (Tensor) - 保留输出,无实际意义。
异常:
TypeError - input_x 不是Tensor。
ValueError - input_x 的shape不是4-D。
- 支持平台:
Ascend
样例:
>>> input_x = Tensor(np.array([[[[1, -2], [-3, 4]], [[-5, 6], [7, -8]]]]), mindspore.float32) >>> relu_v2 = ops.ReLUV2() >>> output, _= relu_v2(input_x) >>> print(output) [[[[1. 0.] [0. 4.]] [[0. 6.] [7. 0.]]]]