skcuda.misc.max

skcuda.misc.max(a_gpu, axis=None, keepdims=False)[source]

Return the maximum of an array or maximum along an axis.

Parameters:
  • a_gpu (pycuda.gpuarray.GPUArray) – Input array
  • axis (int (optional)) – Axis along which the maxima are computed. The default is to compute the maximum of the flattened array.
  • keepdims (bool (optional, default False)) – If True, the axes which are reduced are left in the result as dimensions with size one.
Returns:

out – maximum of elements, or maxima of elements along the desired axis.

Return type:

pycuda.gpuarray.GPUArray or float