2022-11-16 16:28:42 +08:00
|
|
|
|
mindspore.ops.heaviside
|
|
|
|
|
|
========================
|
|
|
|
|
|
|
2022-11-21 15:25:51 +08:00
|
|
|
|
.. py:function:: mindspore.ops.heaviside(x, values)
|
2022-11-16 16:28:42 +08:00
|
|
|
|
|
2022-11-23 16:11:18 +08:00
|
|
|
|
计算输入中每个元素的 Heaviside 阶跃函数。公式定义如下:
|
2022-11-16 16:28:42 +08:00
|
|
|
|
|
|
|
|
|
|
.. math::
|
|
|
|
|
|
\text { heaviside }(\text { x, values })=\left\{\begin{array}{ll}
|
|
|
|
|
|
0, & \text { if x }<0 \\
|
|
|
|
|
|
\text { values, } & \text { if x }==0 \\
|
|
|
|
|
|
1, & \text { if x }>0
|
|
|
|
|
|
\end{array}\right.
|
|
|
|
|
|
|
|
|
|
|
|
参数:
|
|
|
|
|
|
- **x** (Tensor) - 输入Tensor。需为实数类型。
|
2022-11-21 15:25:51 +08:00
|
|
|
|
- **values** (Tensor) - `x` 为零时填充的值。 `values` 可以被 `x` 广播。 `x` 需要与 `values` 数据类型相同。
|
2022-11-16 16:28:42 +08:00
|
|
|
|
|
|
|
|
|
|
返回:
|
|
|
|
|
|
Tensor,数据类型与输入 `x` 和 `values` 相同。
|
|
|
|
|
|
|
|
|
|
|
|
异常:
|
|
|
|
|
|
- **TypeError** - `x` 或 `values` 不是Tensor。
|
|
|
|
|
|
- **TypeError** - `x` 和 `values` 的数据类型不同。
|
|
|
|
|
|
- **ValueError** - 两个输入参数的shape不支持广播。
|