我在尝试学习 Tensorflow 时遇到错误

Question

这是我的代码：

model = keras.models.Sequential([
    keras.layers.Flatten(60000,28,28),
    keras.layers.Dense(128, activation ="relu"),
    keras.layers.Dense(10),
])

这是输出：

TypeError                                 Traceback (most recent call last)
<ipython-input-14-dc603c39e27e> in <cell line: 1>()
      1 model = keras.models.Sequential([
----> 2     keras.layers.Flatten(60000,28,28),
      3     keras.layers.Dense(128, activation ="relu"),
      4     keras.layers.Dense(10),
      5 ])
TypeError: Flatten.__init__() takes from 1 to 2 positional arguments but 4 were given

我可以做什么来解决这个问题？

Answer 1

我认为在第一层，你应该使用

tf.keras.layers.InputLayer

作为输入而不是

tf.keras.layers.Flatten

。

Flatten

用于将多维输入“展平”为一维输入（通常用于从卷积层过渡到完全连接的密集层）。

您的代码应该是：

model = keras.models.Sequential([
    keras.layers.InputLayer(input_shape = (60000,28,28)),
    keras.layers.Dense(128, activation ="relu"),
    keras.layers.Dense(10),
])