Convert einsum computation to dot product to be used in Theano
I have just recently learned about np.einsum and quickly became addicted to it. But it seems that theano doesn't have an equivalent function so I need to convert my numpy code to theano somehow. How can I write the following computation in theano? IX=np.einsum('ijk,lj->ilk',p1,KX) You only need to rearrange your axes to get this to work: >>> 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)) So with that in mind: >>> a = np.random.rand(3, 5, 4) >>> b = np.random.rand(6, 5) >>> out_ein = np.einsum('ijk,lj