- cupy.argmax(a, axis=None, dtype=None, out=None, keepdims=False)¶
Returns the indices of the maximum along an axis.
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.