in tensorflow, I did a tensordot between a X tensor with shape (N, T, D) and a beta with shape (S, D). The idea is to dot alone the D axis on both tensors. in tensorflow this is expressed as
tf.tensordot(X, beta, [[2], [1]])
what would be the equivalent thing in pymc3 / theano?