amax(a, axis=None, out=None, keepdims=False)¶
Returns the maximum of an array or the maximum along an axis.
When at least one element is NaN, the corresponding min value will be NaN.
The maximum of
a, along the axis if specified.
- Return type
When cuTENSOR accelerator is used, the output value might be collapsed for reduction axes that have one or more NaN elements.