forked from mindspore-Ecosystem/mindspore
26 lines
945 B
ReStructuredText
26 lines
945 B
ReStructuredText
|
mindspore.ops.Heaviside
|
|||
|
|
=======================
|
|||
|
|
|
|||
|
|
.. py:class:: mindspore.ops.Heaviside
|
|||
|
|
|
|||
|
|
计算输入中每个元素的Heaviside步长函数。
|
|||
|
|
|
|||
|
|
.. 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，实数类型。
|
|||
|
|
- **values** (Tensor) - 在 `x` 中为0的位置应用其值。 `values` 可以同 `x` 进行广播。 `x` 与 `values` 的数据类型应该相同。
|
|||
|
|
|
|||
|
|
输出：
|
|||
|
|
Tensor，与 `x` 和 `values` 的数据类型相同。
|
|||
|
|
|
|||
|
|
异常：
|
|||
|
|
- **TypeError** - 如果 `x` 或 `values` 不是Tensor。
|
|||
|
|
- **TypeError** - 如果 `x` 与 `values` 的数据类型不一致。
|
|||
|
|
- **ValueError** - 如果两个输入的shape之间无法进行广播。
|