mindspore.mint.nn.functional.relu6

mindspore.mint.nn.functional.relu6(input, inplace=False)[源代码]

逐元素计算输入Tensor的ReLU（修正线性单元），其上限为6。

\[\text{ReLU6}(x) = \min(\max(0,x), 6)\]

返回 \(\min(\max(0,x), 6)\) 元素的值。

ReLU6函数图：

警告

这是一个实验性API，后续可能修改或删除。

参数：

input (Tensor) - 输入Tensor。数据类型为int8、int16、int32、int64、uint8、float16、float32、bfloat16.
inplace (bool, 可选) - 是否采用原地更新模式，默认值为 False。

返回：

Tensor，其shape和数据类型与`input`相同。

异常：

TypeError - input 不是Tensor。
TypeError - input 的数据类型不是int8、int16、int32、int64、uint8、float16、float32、bfloat16。

支持平台：

Ascend

样例：

>>> import mindspore
    >>> import numpy as np
    >>> from mindspore import Tensor, mint
    >>> x = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32)
    >>> result = mint.nn.functional.relu6(x)
    >>> print(result)
    [[0. 4. 0.]
     [2. 0. 6.]]