Implementation of for loop in tensorflow lazy execution mode

Question

I am trying to implement a for loop into tensorflow lazy execution mode. I know that the equivalent for tensorflow is tf.while_loop but it seem cum