TensorFlow / Keras: слой _ не имеет входящих узлов

Question

Я хотел бы получить доступ к output слоя Dense в обратном вызове, но я получаю сообщение об ошибке: AttributeError: Layer dense has no inbound nodes.

Есть другие похожие вопросы, но у меня не получилось ответить. Чтобы решить эту проблему, нужно добавить InputLayer в качестве первого слоя или определить input_shape первого слоя. К сожалению, оба способа не работают в этом примере:

class MyModel(keras.Model):
    def __init__(self):
        super(MyModel, self).__init__()

        self.flatten = keras.layers.Flatten(input_shape=(28, 28, 1))
        self.dense = keras.layers.Dense(64, activation=tf.nn.relu)
        self.classifier = keras.layers.Dense(10, name='classifier')

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

        return self.classifier(x)

Я делаю что-то не так или это обратный вызов, который вызывает эту проблему?