mindspore.mint.nn.Threshold

class mindspore.mint.nn.Threshold(threshold, value, inplace=False)

逐元素计算Threshold激活函数。

Threshold定义为：

\[\begin{split}y = \begin{cases} x, &\text{ if } x > \text{threshold} \\ \text{value}, &\text{ otherwise } \end{cases}\end{split}\]

参数：

• threshold (Union[int, float]) - 阈值。

• value (Union[int, float]) - 小于阈值时的填充值。

• inplace (bool, 可选) - 是否启用原地更新功能。默认 False 。

输入：

• input (Tensor) - 输入tensor。

输出：

Tensor。

支持平台：

Ascend

样例：

>>> import mindspore
>>> inputs = mindspore.tensor([0.0, 2, 3], mindspore.float32)
>>> net = mindspore.mint.nn.Threshold(1, 100)
>>> outputs = net(inputs)
>>> print(outputs)
[100.   2.   3.]