asharma20

I have the same question. I would like some confirmation on this but from what I understand the shapes of `self.word_embedding_output, self.output_embedding_table, self.embedding_output` are with respect to the embedding size....