dpnp.sum

dpnp.sum(a, /, *, axis=None, dtype=None, keepdims=False, out=None, initial=0, where=True)[source]

Sum of array elements over a given axis.

For full documentation refer to numpy.sum.

Parameters:

  • a ({dpnp.ndarray, usm_ndarray}) – Input array.

  • axis (int or tuple of ints, optional) – Axis or axes along which sums must be computed. If a tuple of unique integers, sums are computed over multiple axes. If None, the sum is computed over the entire array. Default: None.

  • dtype (dtype, optional) –

    Data type of the returned array. If None, the default data type is inferred from the “kind” of the input array data type.

    • If a has a real-valued floating-point data type,

      the returned array will have the default real-valued floating-point data type for the device where input array a is allocated.

    • If a has signed integral data type, the returned array

      will have the default signed integral type for the device where input array a is allocated.

    • If a has unsigned integral data type, the returned array

      will have the default unsigned integral type for the device where input array a is allocated.

    • If a has a complex-valued floating-point data type,

      the returned array will have the default complex-valued floating-pointer data type for the device where input array a is allocated.

    • If a has a boolean data type, the returned array will

      have the default signed integral type for the device where input array a is allocated.

    If the data type (either specified or resolved) differs from the data type of a, the input array elements are cast to the specified data type before computing the sum. Default: None.

  • out ({None, dpnp.ndarray, usm_ndarray}, optional) – Alternative output array in which to place the result. It must have the same shape as the expected output, but the type of the output values will be cast if necessary. Default: None.

  • keepdims (bool, optional) – If True, the reduced axes (dimensions) are included in the result as singleton dimensions, so that the returned array remains compatible with the input array according to Array Broadcasting rules. Otherwise, if False, the reduced axes are not included in the returned array. Default: False.

Returns:

out – An array containing the sums. If the sum is computed over the entire array, a zero-dimensional array is returned. The returned array has the data type as described in the dtype parameter description above.

Return type:

dpnp.ndarray

Limitations

Parameters initial and where are only supported with their default values. Otherwise NotImplementedError exception will be raised.

See also

dpnp.ndarray.sum

Equivalent method.

dpnp.cumsum

Cumulative sum of array elements.

dpnp.trapz

Integration of array values using the composite trapezoidal rule.

dpnp.mean

Compute the arithmetic mean.

dpnp.average

Compute the weighted average.

Examples

>>> import dpnp as np
>>> np.sum(np.array([0.5, 1.5]))
array(2.)
>>> np.sum(np.array([0.5, 0.7, 0.2, 1.5]), dtype=np.int32)
array(1)
>>> a = np.array([[0, 1], [0, 5]])
>>> np.sum(a)
array(6)
>>> np.sum(a, axis=0)
array([0, 6])
>>> np.sum(a, axis=1)
array([1, 5])