jax.lax.min#
- jax.lax.min(x, y)[source]#
Elementwise minimum: \(\mathrm{min}(x, y)\)
This function lowers directly to the stablehlo.minimum operation for non-complex inputs. For complex numbers, this uses a lexicographic comparison on the (real, imaginary) pairs.
- Parameters:
x (ArrayLike) – Input arrays. Must have matching dtypes. If neither is a scalar,
x
andy
must have the same rank and be broadcast compatible.y (ArrayLike) – Input arrays. Must have matching dtypes. If neither is a scalar,
x
andy
must have the same rank and be broadcast compatible.
- Returns:
An array of the same dtype as
x
andy
containing the elementwise minimum.- Return type:
See also
jax.numpy.minimum()
: more flexibly NumPy-style minimum.jax.lax.reduce_min()
: minimum along an axis of an array.jax.lax.max()
: elementwise maximum.