Python 子函数变量未保存在tensorflow中

我有一个解码器函数，其中包含一个子函数，该子函数在输入2D张量和2xn形变量之间运行迭代求和。此求和函数在训练期间正常工作，但是函数中使用的变量（2xn形变量和可能的偏差项）未保存在任何模型检查点中。因此，当我加载一个模型进行解码时，函数失败，因为变量不在检查点中。我尝试将函数移出并将其用作普通函数，但仍然没有在检查点中看到变量

---

我怀疑saver.save（）中的默认保存范围可能有问题。

能否添加一些代码示例？