def einsum_twoloop(a, b):
    d = np.zeros(N)
for i in range(N):
    for j in range(i):
        d[i] += np.einsum('ij,ji->', a[:, :, (i)], b[:, :, (j)])
return d
Умножение матриц с зависимостью от итератора — NumPy