How to find out the maximum length of a dimension in a ragged dataset...

tf.data.datasets set each batch (prefetch)...