keras - 如何在keras中仅打印模型最后n层的摘要

Question

model.summary()打印整个模型的细节。有没有办法只打印最后 n 层摘要？

如果没有，我可以从现有预训练模型的最后 n 层创建一个新模型并打印它的摘要。

我尝试了以下方法，但可能由于共享输入而出现错误：

temp_model = Model(inputs=base_model.layers[-4].input, outputs = base_model.layers[-1].output)
print(temp_model.summary())

任何帮助将不胜感激。

标签： keras