Tensorflow:我可以重用keras模型的一部分作为操作或函数吗？

import tensorflow as tf
from tensorflow import keras

# 定义要重用的模型的一部分
class MyModel(tf.keras.Model):
    def __init__(self):
        super(MyModel, self).__init__()
        self.dense = keras.layers.Dense(64, activation='relu')

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

# 创建一个新的模型，并重用MyModel的一部分
class AnotherModel(tf.keras.Model):
    def __init__(self):
        super(AnotherModel, self).__init__()
        self.my_model = MyModel()
        self.dense = keras.layers.Dense(10, activation='softmax')

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

# 使用新模型进行训练和推理
model = AnotherModel()
# ...