A = dstack([[[1, 1], [1, 1]], [[2, 3], [4, 5]], [[6, 7], [8, 9]]])
B = dstack([[[2, 2], [2, 2]], [[10, 11], [12, 13]], [[14, 15], [16, 17]]])
C = einsum('...k,...k', A, B)
Numpy equivalent of dot(A,B,3)