jax.lax.min

jax.lax.min(x, y)

元素级最小值：\(\mathrm{min}(x, y)\)

此函数直接降级到 stablehlo.minimum 操作，用于非复数输入。 对于复数，这使用 (实部, 虚部) 对的字典序比较。

参数:

• x (ArrayLike) – 输入数组。 必须具有匹配的 dtypes。 如果两者都不是标量，则 x 和 y 必须具有相同的秩并且是广播兼容的。

• y (ArrayLike) – 输入数组。 必须具有匹配的 dtypes。 如果两者都不是标量，则 x 和 y 必须具有相同的秩并且是广播兼容的。

返回:

与 x 和 y 具有相同 dtype 的数组，包含元素级最小值。

返回类型:

Array

另请参阅

• jax.numpy.minimum()：更灵活的 NumPy 风格最小值。

• jax.lax.reduce_min()：沿数组轴的最小值。

• jax.lax.max()：元素级最大值。