- jax.numpy.nanmax(a, axis=None, out=None, keepdims=False, initial=None, where=None)#
Return the maximum of an array or maximum along an axis, ignoring any
LAX-backend implementation of
Original docstring below.
NaNs. When all-NaN slices are encountered a
RuntimeWarningis raised and NaN is returned for that slice.
a (array_like) – Array containing numbers whose maximum is desired. If a is not an array, a conversion is attempted.
keepdims (bool, optional) –
If this is set to True, the axes which are reduced are left in the result as dimensions with size one. With this option, the result will broadcast correctly against the original a.
If the value is anything but the default, then keepdims will be passed through to the max method of sub-classes of ndarray. If the sub-classes methods does not implement keepdims any exceptions will be raised.
initial (scalar, optional) – The minimum value of an output element. Must be present to allow computation on empty slice. See ~numpy.ufunc.reduce for details.
where (array_like of bool, optional) – Elements to compare for the maximum. See ~numpy.ufunc.reduce for details.
nanmax – An array with the same shape as a, with the specified axis removed. If a is a 0-d array, or if axis is None, an ndarray scalar is returned. The same dtype as a is returned.
- Return type: