jax.numpy.minimum

jax.numpy.minimum(x, y, /)

返回输入数组的逐元素最小值。

JAX 实现的 numpy.minimum。

参数:

• x (ArrayLike) – 输入数组或标量。

• y (ArrayLike) – 输入数组或标量。 x 和 y 应该具有相同的形状或具有广播兼容性。

返回:

一个包含 x 和 y 逐元素最小值的数组。

返回类型:

Array

注意

对于每对元素，如果两个元素都是有限数字，则 jnp.minimum 返回

• 两者中较小的那个。

• 如果一个元素是 nan，则返回 nan。

另请参阅

• jax.numpy.maximum()：返回输入数组的逐元素最大值。

• jax.numpy.fmin()：返回输入数组的逐元素最小值，忽略 NaN。

• jax.numpy.amin()：返回给定轴上数组元素的最小值。

• jax.numpy.nanmin()：返回给定轴上数组元素的最小值，忽略 NaN。

示例

输入 x.shape == y.shape

>>> x = jnp.array([2, 3, 5, 1])
>>> y = jnp.array([-3, 6, -4, 7])
>>> jnp.minimum(x, y)
Array([-3,  3, -4,  1], dtype=int32)

具有广播兼容性的输入

>>> x1 = jnp.array([[1, 5, 2],
...                 [-3, 4, 7]])
>>> y1 = jnp.array([-2, 3, 6])
>>> jnp.minimum(x1, y1)
Array([[-2,  3,  2],
       [-3,  3,  6]], dtype=int32)

带有 nan 的输入

>>> nan = jnp.nan
>>> x2 = jnp.array([[2.5, nan, -2],
...                 [nan, 5, 6],
...                 [-4, 3, 7]])
>>> y2 = jnp.array([1, nan, 5])
>>> jnp.minimum(x2, y2)
Array([[ 1., nan, -2.],
       [nan, nan,  5.],
       [-4., nan,  5.]], dtype=float32)