tensorflow 2.3 :如何在组合自定义模型后从特定层获得输出

import tensorflow as tf
from tensorflow.keras.models import Model

class MyModel(Model):
    def __init__(self):
        super(MyModel, self).__init__()
        self.conv1 = tf.keras.layers.Conv2D(32, 3, activation='relu')
        self.flatten = tf.keras.layers.Flatten()
        self.dense = tf.keras.layers.Dense(10, activation='softmax')

    def call(self, inputs):
        x = self.conv1(inputs)
        x = self.flatten(x)
        return self.dense(x)

model = MyModel()

input_data = tf.keras.Input(shape=(28, 28, 1))
model(input_data)

layer_output = model.conv1.output