tf.function Cannot take the length of shape with unknown rank

Question

Have tried to decorate a function containing tf.TensorArray with tf.function and worked fine in tf version 2.4.0.

However, when the tf