jax.lax.max

Warning

This page was created from a pull request (#9655).

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 (Any) –

  • y (Any) –

Return type

Any