public sum(axis as Int) as NDArray<T>:
Computes the sum along one axis.
a = NDArray<Int>.fromList([[1, 2], [3, 4]]) print(a.sum(0)) # [4, 6] print(a.sum(1)) # [3, 7]