I only managed to extract one diagonal using Numpy einsum. How do I get the other diagonals like [6, 37, 68, 99] with help of einsum?

```
x =  np.arange(1, 26 ).reshape(5,5)
y =  np.arange(26, 51).reshape(5,5)
z =  np.arange(51, 76).reshape(5,5)
t =  np.arange(76, 101).reshape(5,5)
p =  np.arange(101, 126).reshape(5,5)

a4 = np.array([x, y, z, t, p]

```

Extracting one diagonal:

```
>>>np.einsum('iii->i', a4)
>>>[  1  32  63  94 125]

```