jax.numpy.minimum

目录

jax.numpy.minimum#

jax.numpy.minimum(x, y, /)[源代码][源代码]#

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

JAX implementation of numpy.minimum.

参数:

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

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

返回:

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

返回类型:

Array

备注

对于每一对元素，jnp.minimum 返回：

  • 如果两个元素都是有限数，则返回较小的那个。

  • nan 如果一个元素是 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)
目录