argmax(a_gpu, axis, keepdims=False)[source]¶
Indices of the maximum values along an axis.
- 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.
out – Array of indices into the array.