(a_gpu, axis, keepdims=False)
return _minmax_impl(a_gpu, axis, "max", keepdims=keepdims)[1]
Indices of the maximum values along an axis.

    Parameters
    ----------
    a_gpu : pycuda.gpuarray.GPUArray
        Input array
    axis : int
        Axis along which the maxima are computed.
    keepdims : bool (optional, default False)
        If True, the axes which are reduced are left in the result as
        dimensions with size one.

    Returns
    -------
    out : pycuda.gpuarray.GPUArray
        Array of indices into the array.
Indices of the maximum values along an axis.
Indices of the maximum values along an axis .
def argmax(a_gpu, axis, keepdims=False):
    '''
    Indices of the maximum values along an axis.

    Parameters
    ----------
    a_gpu : pycuda.gpuarray.GPUArray
        Input array
    axis : int
        Axis along which the maxima are computed.
    keepdims : bool (optional, default False)
        If True, the axes which are reduced are left in the result as
        dimensions with size one.

    Returns
    -------
    out : pycuda.gpuarray.GPUArray
        Array of indices into the array.
    '''
    if axis is None:
        raise NotImplementedError("Can't compute global argmax")
    return _minmax_impl(a_gpu, axis, "max", keepdims=keepdims)[1]
def argmax ( a_gpu , axis , keepdims = False ) : if axis is None : raise NotImplementedError ( "Can't compute global argmax" ) return _minmax_impl ( a_gpu , axis , "max" , keepdims = keepdims ) [ 1 ]