Pytorch nn嵌入尺寸大小？

Question

Pytorch 中 nn 嵌入的正确尺寸是多少？我正在做批量训练。

当我得到“形状”时，我对下面代码中“self.embeddings”的尺寸应该是什么感到有点困惑？

self.embeddings = nn.Embedding(vocab_size, embedding_dim)

Answer 1

self.embedding 的形状将是 [sentence_length, batch_size, embedding_dim]

其中sentence_length 是每批输入的长度。