word2vec 向量的 Keras 输入规范

Question

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

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

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

我知道输入层需要 2 个维度，但哪两个维度呢？还是我应该自己将输入形状指定为三维形状？

输入层当前为：

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

Answer 1

输入

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