def f5(A):
    rows, cols = A.shape
N = rows + cols - 1
out = np.zeros(N, A.dtype)
for idx in range(rows):
    out[N - idx - cols:N - idx] += A[idx]
return out[::-1]
Вычислите след матрицы по всем диагоналям