inputs = Input(shape=(128,), name='input_layer') x = Dense(64, activation='sigmoid')(inputs) x = BatchNormalization()(x) x = Dense(32, activation='sigmoid')(x) predictions = Dense(10, activation='softmax', name='output_layer')(x) model = Model(inputs=inputs, outputs=predictions)