当我们想要构建一个模型时，通常会使用tf.keras.Sequential的方法进行层次堆叠，在添加第一层网络结构时，我们要指定模型的input_shape，在这里有一个简便方法：如果数据格式是[num_examples, data_dim1, data_dim2, data_dim3,...]，这样的形式的话，它的input_shape都可以统一写成：x_train.shape[1:]，这种方法虽然简单，不过还是建议用具体的输入数据形状，这样的会更清楚。（eg:input_shape=[28, 28]）
原创文章 46
获赞 49
访问量 2176