tensorly.sum

sum(tensor, axis=None)

Compute the sum of a tensor, optionally along an axis.

Parameters:

tensor : tensor

axis : int, optional

If provided, the sum is computed along this axis.

Returns:

out : scalar or tensor