在 tf.function 模式下存储和访问多个类实例的最佳方式

对于多智能体问题，我正在并行训练多个模型。存储这些模型的最佳方式是什么？目前，我将它们保存在列表中，但这会导致在访问用 @tf.function 装饰的函数中的特定元素时出现问题，出于速度原因，我需要...（有关访问列表元素的问题的更多详细信息：{{3 }})

我已经看过这个 (Select item from list of objects of any type in tensorflow 2.x in tf.function) 但如果之前不知道模型的数量，这似乎很不方便。在 tensorflow 中有没有办法存储任意数量的任何类实例？