使用TensorFlow2.*版本进行构建模型,然后打印模型的结构发现Output Shape为multiple,出现的原因是模型不知道输入数据的格式
三种解决办法:
方法一:使用函数式API
# 指定输入和输出层 model=Model(Input(shape=(None,28,28,1)),outputs)
方法二:在class类中第一个层添加input_shape
self.conv1 = Conv2D(filters=6, kernel_size=5, input_shape=(32, 32, 1))
方法三:调用模型的call函数,参数为输入数据格式的Input层
history.call(Input(shape=(32, 32, 1)))