sum

dragon.math.sum(
  inputs,
  axis=None,
  keepdims=False,
  **kwargs
)[source]

Compute the sum value of elements along the given axis.

axis could be negative or None:

x = dragon.constant([[1, 2, 3], [4, 5, 6]])

# A negative axis is the last-k axis
print(dragon.math.sum(x, 1))
print(dragon.math.sum(x, -1))  # Equivalent

# If axis is None, the vector-style reduction
# will be applied to return a scalar result
print(dragon.math.sum(x))  # 21

# Also, axis could be a sequence of integers
print(dragon.math.sum(x, (0, 1)))  # 21
Parameters:
  • inputs (dragon.Tensor) The input tensor.
  • axis (Union[int, Sequence[int]], optional) The axis to reduce.
  • keepdims (bool, optional, default=False) Keep the reduced dimensions or not.
Returns:

dragon.Tensor The output tensor.