sums main diagonal of A

Call signature
('ii', A)
NumPy equivalent
trace(A)