view main diagonal of A

Call signature
('ii->i', A)
NumPy equivalent
diag(A)