cupy.amax#

cupy.amax(a, axis=None, out=None, keepdims=False)[source]#

Returns the maximum of an array or the maximum along an axis.

Note

When at least one element is NaN, the corresponding min value will be NaN.

Parameters
  • a (cupy.ndarray) – Array to take the maximum.

  • axis (int) – Along which axis to take the maximum. The flattened array is used by default.

  • out (cupy.ndarray) – Output array.

  • keepdims (bool) – If True, the axis is remained as an axis of size one.

Returns

The maximum of a, along the axis if specified.

Return type

cupy.ndarray

Note

When cuTENSOR accelerator is used, the output value might be collapsed for reduction axes that have one or more NaN elements.

See also

numpy.amax()