Solved by vdw in post #2 
              
              
                Your code and comments are not really helpful, so I can only make some half-informed guesses. 
I assume the shape of inputs is (1, 3) since it comes from a array called trigrams. After embeds=self.embeddings(inputs) the shape of embeds is (1, 3, 10) with 10 being the embed_dim. After embeds=embeds.v…