mindspore.ops.heaviside

mindspore.ops.heaviside(input, values)

逐元素进行Heaviside阶跃函数运算。

支持广播。

$$\begin{array}{r}\text{ heaviside }(\text{ input, values })=\{\begin{array}{ll}0,& \text{ if input }<0\\ \text{ values, }& \text{ if input }=0\\ 1,& \text{ if input }>0\end{array}\end{array}$$

参数：

• input (Tensor) - 输入tensor。

• values (Tensor) - input 中元素为0时填充的值。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input = mindspore.tensor([[-2., 0, 3],
...                           [5, -1, 0],
...                           [0, 7, -3]])
>>> values = mindspore.tensor([2, 0.5, 1])
>>> output = mindspore.ops.heaviside(input, values)
>>> print(output)
[[0.  0.5 1. ]
 [1.  0.  1. ]
 [2.  1.  0. ]]
>>> output = mindspore.ops.heaviside(input, mindspore.tensor(0.5))
>>> print(output)
[[0.  0.5 1. ]
 [1.  0.  0.5]
 [0.5 1.  0. ]]
>>> output = mindspore.ops.heaviside(mindspore.tensor(-3.), values)
>>> print(output)
[0. 0. 0.]