jax.Array.max — JAX documentation (original) (raw)
jax.Array.max#
abstract Array.max(axis=None, out=None, keepdims=False, initial=None, where=None)[source]#
Return the maximum of array elements along a given axis.
Refer to jax.numpy.max() for the full documentation.
Parameters:
- self (Array)
- axis (reductions.Axis)
- out (None)
- keepdims (bool)
- initial (ArrayLike | None)
- where (ArrayLike | None)
Return type: