python - Tensorflow 在张量中查找匹配的字符串

Question

我试图找到以train_stepfrom结尾的变量，但如果没有我得到你需要使用的信息，tf.report_uninitialized_variables(),你就无法迭代张量，但我对它的理解不够好。eager execution.tf.map_fn,

这就是我所拥有的：

variables = []
for s, t in zip(tf.report_uninitialized_variables().eval(session=sess), 
                tf.report_uninitialized_variables()):
    if 'train_step' in s:
        variables.append(t)
train_step_init = tf.variables_initializer(variables, name='train_step_init')

标签： pythontensorflowtensor

Answer 1

原来我可以这样做：

variables = []
for i, v in enumerate(tf.global_variables()):
    name = v.name.split(':')[0].encode('ASCII')
    if name in sess.run(tf.report_uninitialized_variables()[0]):
        if b'train_step' in name:
            variables.append(v)

train_step_init = tf.variables_initializer(variables)
sess.run(train_step_init)