def ijk_jlk_to_il(A, B):
res = np.zeros((A.shape[<NUM_LIT:0>], B.shape[<NUM_LIT:1>]))<EOL>[np.add(np.dot(A[:,:,k], B[:,:,k]), res, out=res) for k in range(B.shape[-<NUM_LIT:1>])]<EOL>return res<EOL>
Faster version of einsum einsum('ijk,jlk->il', A,B)