jax.lax.max#

jax.lax.max(x, y)[source]#

Elementwise maximum: \(\mathrm{max}(x, y)\)

For complex numbers, uses a lexicographic comparison on the (real, imaginary) pairs.

Parameters:
  • x (ArrayLike)

  • y (ArrayLike)

Return type:

Array