是否可以在keras中将中间层设置为输出层

Question

我想尝试一下关于自动编码器的想法。 模型是这样的：

input (pictures) - conv2d - pooling - dense - dense(supervised output) - dense - conv - upsampling - output (pictures)

是否可以训练具有dense(supervised output) 和output (pictures) 所需输出的神经网络？换句话说，我想制作一个分类器并返回。

Answer 1

这可以通过 Keras 功能 API (https://keras.io/getting-started/functional-api-guide/) 完成。

一个最小的例子，模型有 2 个输出，一个来自中间层，一个来自最后一层：

import keras
input = keras.layers.Input(shape=(3,))

intermediate = keras.layers.Dense(10)(input)
final_output = keras.layers.Dense(3)(intermediate)

model = keras.Model(inputs=input, outputs=[intermediate, final_output])