python - 如何通过变量/占位符的名称获取引用？

Question

我指的是名字:

tf.placeholder(tf.float32, name='NAME')
tf.get_variable("W", [n_in, n_out],initializer=w_init())

我有几个占位符，我想在不传递引用的情况下从外部函数访问这些占位符，假设存在包含给定名称的占位符，您如何才能获得对它们的引用？ (这都是在图形构建期间，而不是运行时)

我的第二个问题是，无论作用域如何，我如何才能获得包含给定名称的所有变量？

示例:我所有的权重在许多范围内都有名称“W”，我想将它们全部放入一个列表中。我不想手动添加每一个。同样可以用偏差来完成，假设我想做一个直方图。

Answer 1

首先，您可以使用 tf.Graph.get_tensor_by_name() 获取占位符.例如，假设您正在使用默认图表:

placeholder1 = tf.placeholder(tf.float32, name='NAME')
placeholder2 = tf.get_default_graph().get_tensor_by_name('NAME:0')
assert placeholder1 == placeholder2

其次，我将使用以下函数获取具有给定名称的所有变量(无论它们的范围如何):

def get_all_variables_with_name(var_name):
    name = var_name + ':0'
    return [var for var in tf.all_variables() if var.name.endswith(name)]