mindspore.ops.threshold
- mindspore.ops.threshold(input, thr, value)[source]
Returns each element of input after thresholding by thr as a Tensor.
The formula is defined as follows:
\[\begin{split}y = \begin{cases} input, &\text{ if } input > \text{thr} \\ \text{value}, &\text{ otherwise } \end{cases}\end{split}\]- Parameters
- Returns
Tensor, the same shape and data type as the input.
- Raises
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore >>> from mindspore import Tensor, ops >>> inputs = mindspore.Tensor([0.0, 2, 3], mindspore.float32) >>> outputs = ops.threshold(inputs, 1, 100) >>> print(outputs) [100. 2. 3.]