- cupy.argmax(a, axis=None, dtype=None, out=None, keepdims=False)[source]#
Returns the indices of the maximum along an axis.
a (cupy.ndarray) – Array to take argmax.
axis (int) – Along which axis to find the maximum.
ais flattened by default.
dtype – Data type specifier.
out (cupy.ndarray) – Output array.
keepdims (bool) – If
True, the axis
axisis preserved as an axis of length one.
The indices of the maximum of
aalong an axis.
- Return type
keepdimarguments are specific to CuPy. They are not in NumPy.
axisargument accepts a tuple of ints, but this is specific to CuPy. NumPy does not support it.