word2vec 向量的 Keras 输入规范

Keras input specification for word2vec vectors

我阅读了有关此主题的所有其他答案,但我的用例略有不同。

我有一个形状为 (800,128,1) 的 numpy 数组。 800个元素中的每个元素存储一个word2vec嵌入形状(128,1)。现在我想将其作为批量大小为 64 的输入发送到以第一层作为输入层的 Keras 模型。我收到以下错误:

expected party to have 2 dimensions, but got array with shape (800, 128, 1)

我知道输入层需要 2 个维度,但是哪两个?或者我应该自己指定输入形状为三维?

输入层目前是这样的:

Input(shape = (embedding_size, ), name = 'party')

输入的形状是 (embedding_size,),其中 embedding_size 大概是 128。所以输入应该是一个形状为 (batch_size, embedding_size) 的数组,而不是 (batch_size, embedding_size, 1)。您需要重塑数组以省略大小为 1 的最后一个维度。