pytorch nn嵌入尺寸尺寸？

Question

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

我只是对下面代码中“ self.embeddings”的尺寸应该是当我得到“ shape”时的尺寸感到困惑？

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

Answer 1

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

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