I need to pass multiple and different positional arguments to tf.while_loop function, which means arguments of cond are different from arguments of body.
For better under