import numpy as np
a = np.random.rand(3, 4, 5)
b = np.random.rand(5, 6)
np.allclose(np.einsum('ikj,jl->ikl', a, b), np.dot(a, b))
Преобразуйте вычисление einsum в скалярное произведение для использования в Theano.