你如何在 TensorFlow v2 中传递会话对象？

Question

我有一个函数 change_weight() 可以修改任何给定模型中的权重。此函数位于不同的 python 文件中。

因此，如果我有一个简单的神经网络来对 MNIST 图像进行分类，我会在调用此函数之前和之后测试准确性，我发现它可以正常工作。这在 TensorFlow v1 中很容易做到，因为我只需要在函数调用中传递 Session sess 对象，我就可以在另一个文件中获得该会话的权重。

在 TensorFlow v2 中使用 eager execution，我该怎么做？我不再有 Session 对象了。我传递什么？

Answer 1

我能够通过传递 Model 对象并在另一个函数中通过 model.trainable_variables 获取权重来做到这一点。