Передача комбинации структуры данных python и тензоров tf в качестве переменных tf. while_loop

Question

Мне нужно использовать tf.while_loop для моего проекта.Можно ли передать список как словаря питона, так и тензоров тензорного потока в виде списка переменных для функций body и cond?

 a = {'state': 1 , 'path': 2}
 b = tf.get_variable(name = 'my_variable' , shape=[batch_size, None])
 state = {
    'x' : a ,
    'y' : b } 
shape = { 
    'y' : tf.TensorShape([batch_size, None])
   }
 tf.while_loop(
        cond, body, loop_vars=[state],
        shape_invariants=[shape], parallel_iterations=1, back_prop=False)