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

Question

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

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

在 TensorFlow v2 中使用 Eager Execution，我该怎么做？我没有 Session 对象了。我要通过什么？

Answer 1

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