keras获取中间层输出

https://keras-cn.readthedocs.io/en/latest/for_beginners/FAQ/#intermediate_layer

from keras import backend as K
get_3rd_layer_output = K.function([model.layers[0].input, K.learning_phase()],
                                  [model.layers[5].output])
X = x_test
# output in test mode = 0
layer_output = get_3rd_layer_output([X, 0])[0]
print(layer_output.shape)

# output in train mode = 1
layer_output = get_3rd_layer_output([X, 1])[0]