sum#
Signature#
nabla.sum(arg: 'Array', axes: 'int | list[int] | tuple[int, ...] | None' = None, keep_dims: 'bool' = False) -> 'Array'
Description#
sum array elements over given axes.
Examples#
import nabla as nb
# Sum along axes
x = nb.array([[1, 2], [3, 4]])
result = nb.sum(x) # Sum all elements
print(result) # 10
result_axis0 = nb.sum(x, axis=0) # Sum along rows
print(result_axis0) # [4, 6]