JaxBackend.sum

JaxBackend.sum(*args, **kwargs)

Compute the sum of array elements.

Parameters:

arrBackendArray

Input data.

axisint or tuple of ints, optional

Axis or axes to perform the operation on. Default is all.

Returns:

Union[BackendArray, Scalar]

Sum of arr.