skcuda.misc.mean

skcuda.misc.mean(x_gpu, axis=None, out=None, keepdims=False)[source]

Compute the arithmetic means along the specified axis.

Parameters:
  • x_gpu (pycuda.gpuarray.GPUArray) – Array containing numbers whose mean is desired.
  • axis (int (optional)) – Axis along which the means are computed. The default is to compute the mean of the flattened array.
  • out (pycuda.gpuarray.GPUArray (optional)) – Output array in which to place the result.
  • keepdims (bool (optional, default False)) – If True, the axes which are reduced are left in the result as dimensions with size one.
Returns:

out – mean of elements, or means of elements along the desired axis.

Return type:

pycuda.gpuarray.GPUArray