单输入--单输出
input = tf.keras.Input(shape = (256,256,3))
c1 = tf.keras.layers.Conv2D(16,3,activation = 'relu')(input)
c1 = tf.keras.layers.MaxPool2D(3,2)(c1)
c1 = tf.keras.layers.Flatten()(c1)
output = tf.keras.layers.Dense(10,activation = 'softmax')(c1)
model = tf.keras.Model(inputs = input,outputs = output)
tf.keras.utils.plot_model(model,show_shapes=True)
单输入--多输出
方法1:
input = tf.keras.Input(shape = (256,256,3)) c1 = tf.keras.layers.Conv2D(16,3,activation = 'relu')(input) c1 = tf.keras.layers.MaxPool2D(3,2)(c1) c1 = tf.keras.layers.Flatten()(c1) output1 = tf.keras.layers.Dense(10,activation = 'softmax')(c1) output2 &