如何在tensorflow中设置keras子类模型的输入？

import tensorflow as tf
from tensorflow.keras.layers import Dense

class MyModel(tf.keras.Model):
    def __init__(self):
        super(MyModel, self).__init__()
        self.dense = Dense(units=10, activation='relu')

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

input_shape = (None, 20)  # 假设输入特征的形状为(None, 20)
model = MyModel()

model.build(input_shape)

input_data = tf.random.normal((32, 20))  # 假设有32个样本
output = model(input_data)