当我将numpy数组作为输入传递给keras层时，它具有不同的形状

Posted 2023-03-12

【中文标题】当我将numpy数组作为输入传递给keras层时，它具有不同的形状

【英文标题】：numpy array has different shape when I pass it as input to a keras layer

【发布时间】：2018-11-16 09:08:17

【问题描述】：

我有一个这样构建的 keras 编码器（自动编码器的一部分）：

input_vec = Input(shape=(200,))
encoded = Dense(20, activation='relu')(input_vec)
encoder = Model(input_vec, encoded)

我想使用 numpy 生成一个虚拟输入。

>>> np.random.rand(200).shape
(200,)

但如果我尝试将其作为输入传递给编码器，我会收到 ValueError：

>>> encoder.predict(np.random.rand(200))
>>> Traceback (most recent call last):
  File "<console>", line 1, in <module>
  File "/home/francesco/PycharmProjects/W2VAutoencoded/venv/lib/python3.6/site-packages/keras/engine/training.py", line 1817, in predict
    check_batch_axis=False)
  File "/home/francesco/PycharmProjects/W2VAutoencoded/venv/lib/python3.6/site-packages/keras/engine/training.py", line 123, in _standardize_input_data
    str(data_shape))
ValueError: Error when checking : expected input_1 to have shape (200,) but got array with shape (1,)

我错过了什么？

【参考方案1】：

Keras Layers（Input、Dense 等）将单个样本的形状作为参数，Model.predict() 将批处理数据（即堆叠在第一维上的样本）作为输入.

现在，您的模型认为您正在向其传递一批 200 形状为 (1,) 的样本。

这可行：

batch_size = 1
encoder.predict(np.random.rand(batch_size, 200))