def ij_jlk_to_ilk(A, B):
return A.dot(B.reshape(B.shape[<NUM_LIT:0>], -<NUM_LIT:1>)).reshape(A.shape[<NUM_LIT:0>], B.shape[<NUM_LIT:1>], B.shape[<NUM_LIT:2>])<EOL>
Faster version of einsum 'ij,jlk->ilk'