2022-12-24 16:42:54 +08:00
|
|
|
|
mindspore.ops.threshold
|
|
|
|
|
|
=======================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:function:: mindspore.ops.threshold(input_x, thr, value)
|
|
|
|
|
|
|
2023-02-06 16:55:54 +08:00
|
|
|
|
将使用 `thr` 参数对 `input_x` 逐元素阈值化后的结果作为Tensor返回。
|
2022-12-24 16:42:54 +08:00
|
|
|
|
|
|
|
|
|
|
threshold定义为：
|
|
|
|
|
|
|
|
|
|
|
|
.. math::
|
|
|
|
|
|
y =
|
|
|
|
|
|
\begin{cases}
|
|
|
|
|
|
x, &\text{ if } x > \text{thr} \\
|
|
|
|
|
|
\text{value}, &\text{ otherwise }
|
|
|
|
|
|
\end{cases}
|
|
|
|
|
|
|
|
|
|
|
|
参数：
|
|
|
|
|
|
- **input_x** (Tensor) - 输入Tensor，数据类型为float16或float32。
|
|
|
|
|
|
- **thr** (Union[int, float]) - 阈值。
|
|
|
|
|
|
- **value** (Union[int, float]) - 输入Tensor中element小于阈值时的填充值。
|
|
|
|
|
|
|
|
|
|
|
|
返回：
|
|
|
|
|
|
Tensor，数据类型和shape与 `input_x` 的相同。
|
|
|
|
|
|
|
|
|
|
|
|
异常：
|
|
|
|
|
|
- **TypeError** - `input_x` 不是Tensor。
|
|
|
|
|
|
- **TypeError** - `thr` 不是浮点数或整数。
|
|
|
|
|
|
- **TypeError** - `value` 不是浮点数或整数。
|
