如何为另一个模型中的模型设置变量？

import tensorflow as tf

# 定义主模型
class MainModel(tf.keras.Model):
    def __init__(self):
        super(MainModel, self).__init__()
        self.sub_model = SubModel()

    def call(self, inputs):
        # 使用子模型对象设置变量
        self.sub_model.set_variable(inputs)
        # 其他操作...

# 定义子模型
class SubModel(tf.keras.Model):
    def __init__(self):
        super(SubModel, self).__init__()
        self.variable = None

    def set_variable(self, value):
        self.variable = value

    def call(self, inputs):
        # 其他操作...
        pass

# 创建主模型对象和子模型对象
main_model = MainModel()
sub_model = SubModel()

# 设置变量
main_model(sub_model)

# 使用变量
print(sub_model.variable)