jax.Array.max

Contents

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:

Array

Contents