Tensorflow находит подходящие строки в тензоре

Question

Я пытаюсь найти переменные, которые заканчиваются на train_step из 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')

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)