如何在tensorflow 2.0中按名称查找变量 [英] How to find a variable by name in tensorflow2.0

问题描述

TensorFlow 2.0似乎删除了函数tf.global_Variables()。我的问题是：如何在模型中找到具有给定名称的变量？TensorFlow 2.0是否提供了执行此操作的函数？

目前，我们的培训框架无法访问构建模型的代码。它使用tf.global_Variables()在加载的模型中查找各种变量，以推入和拉出训练数据或结果。TensorFlow 2.0能够支持这种框架吗？

推荐答案

TensorFlow 2.0能够支持这种框架吗？

不，不会的。

TensorFlow 2.0的首要设计选择之一是完全删除全局集合(例如tf.global_variables())，转而采用以Python为中心的方法，即使用携带其自身变量的Python对象。

现在推荐的做法是构建一个Kera模型，并通过使用Python对象在代码中的任何地方使用相同的模型。 您可以通过访问model.trainable_variables和model.non_trainable_weights来访问对象的可训练变量和不可训练变量。

简而言之，您必须开始将代码库迁移到面向对象的方法，因为TensorFlow的1.x方法(依赖于全局集合、按名称获取变量等)在2.0中将不可用。

这篇关于如何在tensorflow 2.0中按名称查找变量的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！