Storing Trainable Values in TensorArray for While_loop - tensorflow

Question

I\'m working on a small Tensorflow model using Tensorflow 2.3.0. in the model I use several tf.while_loops and TensorArray. The model is not working as expected. I tried to tro