jax.numpy.heaviside

目录

jax.numpy.heaviside#

jax.numpy.heaviside(x1, x2, /)[源代码][源代码]#

计算 Heaviside 阶跃函数。

LAX-backend 对 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 or scalar

引用

目录