I assume the error is thrown in these lines of code:
for i,e in enumerate(self.all_embeddings):
    embeddings.append(e(x_categorical[:,i]))

Could you check the shape of self.all_embeddings?
Based on your input, if should be only 1, to be able to index x_categorical at dim1.