tf.TensorArray和tf.while_loop组合使用