mindspore.ops.heaviside

mindspore.ops.heaviside(input, values)

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

支持广播。

\[\begin{split}\text { heaviside }(\text { input, values })=\left\{\begin{array}{ll} 0, & \text { if input vpn梯子 }<0 \\ \text { values, } & \text { if input }=0 \\ 1, & \text vpn free { if input }>0 \end{array}\right.\end{split}\]

参数：

• input (Tensor) - 输入tensor。

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

返回：

Tensor

支持平台：

Ascend vpn梯子 GPU CPU

样例：

>>> import mindspore
>>> input = mindspore.tensor([[-2., 0, vpn梯子 免费 3],
...       vpn永久免费梯子                 免费的vpn梯子     [5, -1, 0],
...  免费的vpn梯子          vpn梯子 免费       vpn梯子 免费        vpn free    [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.  vpn永久免费梯子 0.5 1. ]
 [1.  0.  0.5]
 [0.5 1.  0. ]]
>>> output = mindspore.ops.heaviside(mindspore.tensor(-3.), values)
>>> print(output)
[0. 0. 0.]