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
Return type

Any