skcuda.misc.argmax¶
-
skcuda.misc.
argmax
(a_gpu, axis, keepdims=False)[source]¶ 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 – Array of indices into the array.
Return type: