jax.numpy.heaviside

内容

jax.numpy.heaviside(x1, x2, /)[source]#

计算 Heaviside 阶跃函数。

LAX 后端实现 numpy.heaviside()。

原始文档字符串如下。

Heaviside 阶跃函数 [1] 定义为

                      0   if x1 < 0
heaviside(x1, x2) =  x2   if x1 == 0
                      1   if x1 > 0

其中 x2 通常取值为 0.5，但有时也使用 0 和 1。

参数：

x1 (array_like) – 输入值。
x2 (array_like) – 当 x1 为 0 时函数的值。如果 x1.shape != x2.shape，则它们必须可广播到一个共同的形状（这将成为输出的形状）。

返回值：

out – 输出数组，x1 的逐元素 Heaviside 阶跃函数。如果 x1 和 x2 都是标量，则这是一个标量。

返回类型：

ndarray 或标量

参考文献