def argmax(
        self,
        axis: Any = None,
        out: ndarray | None = None,
        keepdims: bool = False,
    ) -> ndarray:
        """a.argmax(axis=None, out=None)

        Return indices of the maximum values along the given axis.

        Refer to :func:`cupynumeric.argmax` for full documentation.

        See Also
        --------
        cupynumeric.argmax : equivalent function

        Availability
        --------
        Multiple GPUs, Multiple CPUs

        """
        if out is not None and out.dtype != np.int64:
            raise ValueError("output array must have int64 dtype")
        if axis is not None and not isinstance(axis, int):
            raise ValueError("axis must be an integer")
        return perform_unary_reduction(
            UnaryRedCode.ARGMAX,
            self,
            axis=axis,
            res_dtype=np.dtype(np.int64),
            out=out,
            keepdims=keepdims,
        )
a.argmax(axis=None, out=None)

        Return indices of the maximum values along the given axis.

        Refer to :func:`cupynumeric.argmax` for full documentation.

        See Also
        --------
        cupynumeric.argmax : equivalent function

        Availability
        --------
        Multiple GPUs, Multiple CPUs

        
cupynumeric/_array/array.py